Vectorize dc and dmc (#3764)

bcgov · Jul 5, 2024 · f08acb8 · f08acb8
1 parent 158fbe0
commit f08acb8
Show file tree

Hide file tree

Showing 6 changed files with 3,664 additions and 8 deletions.
diff --git a/api/app/auto_spatial_advisory/sfms.py b/api/app/auto_spatial_advisory/sfms.py
@@ -1,4 +1,6 @@
-from cffdrs import bui
+from cffdrs import bui, dc, dmc
 from numba import vectorize
 
-vectorized_bui = vectorize(bui)
+vectorized_bui = vectorize(bui)
+vectorized_dc = vectorize(dc)
+vectorized_dmc = vectorize(dmc)
diff --git a/api/app/tests/sfms/test_sfms_dc_vectorized.py b/api/app/tests/sfms/test_sfms_dc_vectorized.py
diff --git a/api/app/tests/sfms/test_sfms_dmc_vectorized.py b/api/app/tests/sfms/test_sfms_dmc_vectorized.py
diff --git a/api/app/tests/utils/numpy_utils.py b/api/app/tests/utils/numpy_utils.py
@@ -0,0 +1,5 @@
+import numpy as np
+
+
+def create_numpy_array(value) -> np.ndarray:
+    return np.array([[value, value, value], [value, value, value], [value, value, value]])
diff --git a/api/poetry.lock b/api/poetry.lock
diff --git a/api/pyproject.toml b/api/pyproject.toml
@@ -44,8 +44,8 @@ httpx = "^0.27.0"
 orjson = "^3.9.0"
 greenlet = "^3.0.0"
 sentry-sdk = {extras = ["fastapi"], version = "^2.0.1"}
-cffdrs = {version = "^0.1.0", source = "psu"}
 numba = "^0.59.1"
+cffdrs = {version = "^0.1.1", source = "psu"}
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.2.1"
@@ -103,4 +103,5 @@ per-file-ignores = { "alembic/versions/00df3c7b5cba_rethink_classification.py" =
     "F811",
 ] }
 line-length = 185
-ignore = ["E712"]
+ignore = ["E712", "F401"]
+