refactor: update function signatures to accept both str and Path types

[Ayyanaruto]

Description

This pull request enhances type flexibility across multiple files by replacing str with Union[str, Path] for file path parameters. This change enables functions to accept both string-based paths and Path objects from the pathlib module, improving compatibility and usability. The pathlib.Path module has also been imported wherever necessary.

Related Issue

Fixes #49

Linked Pruna Pro PR: https://github.com/PrunaAI/prunatree/pull/129

Type of Change

• Added type flexibility to support both Path and str for file-related parameters in save and load functions
• Modified type annotations in files such as save.py, load.py, smash_config.py, and utils.py

How Has This Been Tested?

• Unit tests run:

  • test_save.py
  • test_load.py

Checklist

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes

[davidberenstein1957]

@Ayyanaruto , thanks for the great helpt. Could you add tests for this?

[Ayyanaruto]

Hey @davidberenstein1957 sure 😄 !

[davidberenstein1957]

Hii looks good! Would you be able to use the | operator instead of the Union? Additionally, could you add something like test_deprecation_* to cover the old behavior ?

[Ayyanaruto]

Hii looks good! Would you be able to use the | operator instead of the Union? Additionally, could you add something like test_deprecation_* to cover the old behavior ?

Sure @davidberenstein1957 , Can you elaborate what you mean by > "Additionally, could you add something like test_deprecation_* to cover the old behavior" , By old behaviour you mean where we were using paths type as str

[davidberenstein1957]

@Ayyanaruto there is a tiny error where we should convert the path to string before passing it to the underlying method. Could you fix that?

[Ayyanaruto]

@Ayyanaruto there is a tiny error where we should convert the path to string before passing it to the underlying method. Could you fix that?

Hey @davidberenstein1957, sure! I’m in college at the moment, so I’ll need a little time. I’ll get it fixed as soon as I’m free.

[Ayyanaruto]

Hi @davidberenstein1957 Apologies for the delay I was a bit busy today. Thank you for always reviewing my PRs and for your continued help and support

[davidberenstein1957]

LFTM

[davidberenstein1957]

Hi @Ayyanaruto, could you rebase on main? Somethign like the following should work.

• git checkout main
• git pull
• git checkout feat/pathlib-support-save-load
• git rebase main and resolve conflicts
• git push --force

[Ayyanaruto]

Oh thanks, sorry for responding late