types(mock): Set dtype arg to get right types

Jacob-Stevens-Haas · Feb 18, 2024 · 5369b2d · 5369b2d
1 parent 69a65b6
commit 5369b2d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -92,7 +92,7 @@ addopts = '-m "not slow"'
 markers = ["slow"]
 
 [tool.mypy]
-files = ["src/gen_experiments/__init__.py"]
+files = ["src/gen_experiments/__init__.py", "src/gen_experiments/utils.py"]
 
 [[tool.mypy.overrides]]
 module="auto_ks.*"

diff --git a/src/gen_experiments/__init__.py b/src/gen_experiments/__init__.py
@@ -11,7 +11,7 @@
 from .utils import SINDyTrialData
 
 this_module = importlib.import_module(__name__)
-BORING_ARRAY = np.ones((2, 2))
+BORING_ARRAY = np.ones((2, 2), dtype=float)
 
 Scores = Mapping[str, float]
 
@@ -51,7 +51,7 @@ def run(
                 # "coefficients": boring_array,
                 "feature_names": ["1"],
                 "input_features": ["x", "y"],
-                "t_train": np.arange(0, 1, 1),
+                "t_train": np.arange(0, 1, 1, dtype=float),
                 "x_train": BORING_ARRAY,
                 "x_true": BORING_ARRAY,
                 "smooth_train": BORING_ARRAY,