chore: torch and pytorch-lightning as optional dependencies

.pre-commit-config.yaml

@@ -3,32 +3,32 @@
 default_stages: [commit]
 repos:
   - repo: https://github.com/compilerla/conventional-pre-commit
-    rev: v2.1.1
+    rev: v2.2.0
     hooks:
       - id: conventional-pre-commit
         stages: [commit-msg]
   - repo: https://github.com/charliermarsh/ruff-pre-commit
-    rev: "v0.0.259"
+    rev: 'v0.0.262'
     hooks:
       - id: ruff
         args: [--fix, --exit-non-zero-on-fix]
   - repo: https://github.com/psf/black
-    rev: 23.1.0
+    rev: 23.3.0
     hooks:
       - id: black-jupyter
   - repo: https://github.com/PyCQA/docformatter
-    rev: v1.5.1
+    rev: v1.6.3
     hooks:
       - id: docformatter
         additional_dependencies: [tomli]
         args: ["--in-place", "--config", "./pyproject.toml"]
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.1.1
+    rev: v1.2.0
     hooks:
       - id: mypy
         additional_dependencies: ["types-requests"]
   - repo: https://github.com/pdm-project/pdm
-    rev: 2.5.2
+    rev: 2.5.3
     hooks:
       - id: pdm-lock-check
       - id: pdm-export

CHANGELOG.md

@@ -32,6 +32,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Change embedders and joiners interface to have `.transform` method
 - Change linter to Ruff and removed flake8, isort, pydocstyle
 - Change default value inside `transform` function of IntersectionJoiner to not return geometry.
+- Make torch and pytorch-lightning as optional dependencies ([#210](https://github.com/srai-lab/srai/issues/210))
 
 ### Deprecated
 

docs/getting_started/README.md

@@ -19,6 +19,7 @@ The following optional dependencies can be installed to enable additional functi
 * `srai[voronoi]` - dependencies to use Voronoi-based regionization method
 * `srai[gtfs]` - dependencies to process GTFS data
 * `srai[plotting]` - dependencies to plot graphs and maps
+* `srai[torch]` - dependencies to use torch-based embedders
 
 ## Usage