Add all dependencies to pyproject.toml. Remove conda environments.

swisstopo · May 31, 2024 · dd33a21 · dd33a21
1 parent f575b6d
commit dd33a21
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 67 deletions.
diff --git a/environment-dev.yml b/environment-dev.yml
diff --git a/environment-prod.yml b/environment-prod.yml
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,18 +12,37 @@ dependencies = [
     "boto3",
     "pandas",
     "levenshtein",
-    "pathlib==1.0.1",
-    "python-dotenv==1.0.1",
-    "scikit-learn==1.4.0",
-    "click==8.1.7",
-    "PyYAML==6.0.1",
-    "langdetect==1.0.9",
+    "pathlib",
+    "python-dotenv",
+    "scikit-learn>=1.4.0",
+    "click>=8.0.0",
+    "PyYAML>=6.0.1",
+    "langdetect>=1.0.9",
+    "regex",
+    "amazon-textract-textractor",
     "backoff",
-    "PyMuPDF==1.23.26",
+    "PyMuPDF>=1.23.26",
     "opencv-python-headless",
-    "quads==1.1.0",
+    "quads>=1.1.0",
 ]
 
+[project.optional-dependencies]
+test = [
+    "pytest==8.1.1",
+    "pytest-cov==5.0.0"
+]
+lint = [
+    "pre-commit==3.6.2",
+]
+experiment_tracking = [
+    "mlflow==2.11.1"
+]
+visualize = [
+    "matplotlib==3.8.0"
+]
+
+all = ["swisstopo-boreholes[test, lint, experiment_tracking, visualize]"]
+
 [project.scripts]
 boreholes-extract-layers = "stratigraphy.main:click_pipeline"
 boreholes-download-profiles = "stratigraphy.get_files:download_directory_froms3"