fixes to test

DeplanckeLab · May 23, 2024 · a08f801 · a08f801
1 parent fddb7f1
commit a08f801
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,6 +31,7 @@ dependencies = [
     "tensorstore",
     "requests",
     "zarr",
+    "pathlibfs",
 ]
 dynamic = ["version", "readme"]
 license = {text = "MIT"}

diff --git a/src/chromatinhd/models/diff/model/binary.py b/src/chromatinhd/models/diff/model/binary.py
@@ -178,8 +178,6 @@ def train_model(
         Trains the model
         """
 
-        # print(regions_oi, n_epochs)
-
         if fragments is None:
             fragments = self.fragments
         if clustering is None:
@@ -515,7 +513,7 @@ def train_models(
                     path=model_folder,
                     **self.model_params,
                 )
-                model.train_model(device=device, pbar=True, regions_oi=regions_oi, **self.train_params)
+                model.train_model(device=device, pbar=True, regions_oi=regions_oi, **{**self.train_params, **kwargs})
                 model.save_state()
 
                 model = model.to("cpu")

diff --git a/tests/data/motifscan/test_motifscan.py b/tests/data/motifscan/test_motifscan.py
@@ -63,7 +63,7 @@ def test_scan():
         scores, positions, strands = chd.data.motifscan.motifscan.scan(onehot, pwm, cutoff=1.5)
 
         assert np.allclose(scores, test["expected_scores"]), (test["sequence"], scores)
-        assert np.allclose(positions, test["expected_positions"].astype(np.int))
+        assert np.allclose(positions, test["expected_positions"].astype(positions.dtype))
         assert np.allclose(strands, test["expected_strands"].astype(np.int8))