build: add ultralytics and sam as deps

GIScience · Nov 14, 2023 · e68f1b3 · e68f1b3
1 parent 4c46911
commit e68f1b3
Show file tree

Hide file tree

Showing 3 changed files with 165 additions and 1 deletion.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,6 +28,8 @@ pyproj = "^3.6.1"
 neptune = "^1.8.3"
 torch = {version = "^2.1.0+cpu", source = "pytorch"}
 torchvision = {version = "^0.16.0+cpu", source = "pytorch"}
+ultralytics = "^8.0.209"
+segment-anything = {git = "https://github.com/facebookresearch/segment-anything.git"}
 
 [tool.poetry.group.dev.dependencies]
 # Versions are fixed to match versions used by pre-commit

diff --git a/tests/unit/test_depedencies.py b/tests/unit/test_depedencies.py
@@ -0,0 +1,4 @@
+def test_torch():
+    import torch
+
+    torch.rand(5, 3)