chore: Update tests

Author: kshitij12345

.github/workflows/ci.yml

@@ -23,7 +23,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest, macos-latest, windows-latest]
-        python-version: [3.9]
+        python-version: [3.9, 3.13]
     steps:
       - uses: actions/checkout@v2
 

.pre-commit-config.yaml

@@ -1,22 +1,22 @@
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.3.0
+    rev: v5.0.0
     hooks:
       - id: check-yaml
       - id: end-of-file-fixer
       - id: trailing-whitespace
   - repo: https://github.com/pycqa/isort
-    rev: 5.12.0
+    rev: 6.0.1
     hooks:
       - id: isort
         args: [--settings-path, pyproject.toml]
   - repo: https://github.com/psf/black
-    rev: 22.6.0
+    rev: 25.1.0
     hooks:
       - id: black
         args: [--config, pyproject.toml]
   - repo: https://github.com/pycqa/flake8
-    rev: 5.0.4
+    rev: 6.0.0
     hooks:
       - id: flake8
         entry: pflake8

CHANGELOG.md

@@ -1,4 +1,12 @@
-## neptune-pytorch 1.1.0
+## neptune-pytorch 2.0.0
+
+### Changes
+- Rename `save_model` to `log_model` and `save_checkpoint` to `log_checkpoint`. (https://github.com/neptune-ai/neptune-pytorch/pull/9)
+- Prefix private methods with underscore. (https://github.com/neptune-ai/neptune-pytorch/pull/12)
+- Add docstrings for `log_model` and `log_checkpoint`. (https://github.com/neptune-ai/neptune-pytorch/pull/11)
+
+
+## neptune-pytorch 1.1.0 (YANKED)
 
 ### Fixes
 - Rename `save_model` to `log_model` and `save_checkpoint` to `log_checkpoint`. (https://github.com/neptune-ai/neptune-pytorch/pull/9)

pyproject.toml

@@ -9,10 +9,11 @@ style = "semver"
 pattern = "default-unprefixed"
 
 [tool.poetry.dependencies]
-python = "^3.7"
+python = "^3.9"
 
 # Python lack of functionalities from future versions
 importlib-metadata = { version = "*", python = "<3.8" }
+setuptools = { version = "*", python = "<3.8" }
 
 torch = ">1.8.0"
 
@@ -25,28 +26,21 @@ neptune = { version = ">=1.0.0", optional = true }
 torchviz = { version = "*", optional = true }
 
 [tool.poetry.extras]
-dev = [
-    "pre-commit",
-    "pytest",
-    "pytest-cov",
-    "pydot",
-    "neptune",
-    "torchviz",
-]
+dev = ["pre-commit", "pytest", "pytest-cov", "pydot", "neptune", "torchviz"]
 
 [tool.poetry]
 authors = ["neptune.ai <[email protected]>"]
 description = "Neptune.ai pytorch integration library"
 repository = "https://github.com/neptune-ai/neptune-pytorch"
 homepage = "https://neptune.ai/"
-documentation = "https://docs.neptune.ai/integrations-and-supported-tools/model-training/pytorch"
+documentation = "https://docs-beta.neptune.ai"
 include = ["CHANGELOG.md"]
 license = "Apache License 2.0"
 name = "neptune-pytorch"
 readme = "README.md"
 version = "0.0.0"
 classifiers = [
-    "Development Status :: 4 - Beta",
+    "Development Status :: 5 - Production/Stable",
     "Environment :: Console",
     "Intended Audience :: Developers",
     "Intended Audience :: Science/Research",
@@ -67,17 +61,15 @@ keywords = [
     "ML Model Store",
     "ML Metadata Store",
 ]
-packages = [
-    { include = "neptune_pytorch", from = "src" },
-]
+packages = [{ include = "neptune_pytorch", from = "src" }]
 
 [tool.poetry.urls]
 "Tracker" = "https://github.com/neptune-ai/neptune-pytorch/issues"
-"Documentation" = "https://docs.neptune.ai/integrations-and-supported-tools/model-training/pytorch"
+"Documentation" = "https://docs-beta.neptune.ai"
 
 [tool.black]
 line-length = 120
-target-version = ['py37', 'py38', 'py39', 'py310']
+target-version = ['py39', 'py310', 'py311', 'py312', 'py313']
 include = '\.pyi?$'
 exclude = '''
 /(

src/neptune_pytorch/impl/__init__.py

@@ -113,19 +113,19 @@ def __init__(
         self._vis_hook_handler = None
         if log_model_diagram:
             self.run[self._base_namespace]["model"]["summary"] = str(model)
-            self.add_visualization_hook()
+            self._add_visualization_hook()
 
         self.log_gradients = log_gradients
         self._gradients_iter_tracker = {}
         self._gradients_hook_handler = {}
         if self.log_gradients:
-            self.add_hooks_for_grads()
+            self._add_hooks_for_grads()
 
         self.log_parameters = log_parameters
         self._params_iter_tracker = 0
         self._params_hook_handler = None
         if self.log_parameters:
-            self.add_hooks_for_params()
+            self._add_hooks_for_params()
 
         # Log integration version
         root_obj = self.run
@@ -134,7 +134,7 @@ def __init__(
 
         root_obj[INTEGRATION_VERSION_KEY] = __version__
 
-    def add_hooks_for_grads(self):
+    def _add_hooks_for_grads(self):
         for name, parameter in self.model.named_parameters():
             self._gradients_iter_tracker[name] = 0
 
@@ -145,7 +145,7 @@ def hook(grad, name=name):
 
             self._gradients_hook_handler[name] = parameter.register_hook(hook)
 
-    def add_visualization_hook(self):
+    def _add_visualization_hook(self):
         if not IS_TORCHVIZ_AVAILABLE:
             msg = "Skipping model visualization because no torchviz installation was found."
             warnings.warn(msg)
@@ -170,7 +170,7 @@ def hook(module, input, output):
 
         self._vis_hook_handler = self.model.register_forward_hook(hook)
 
-    def add_hooks_for_params(self):
+    def _add_hooks_for_params(self):
         def hook(module, inp, output):
             self._params_iter_tracker += 1
             if self._params_iter_tracker % self.log_freq == 0: