all tests passing

Genentech · Jul 26, 2024 · 238f0c3 · 238f0c3
1 parent bc737ed
commit 238f0c3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ requires = [
 [project]
 authors = [{ email = "[email protected]", name = "Allen Goodman" }]
 dependencies = [
-    "optree",
+    "optree==0.11.0",
     "pooch",
     "torch==2.2.2",
     "tqdm",

diff --git a/src/beignet/func/_interact.py b/src/beignet/func/_interact.py
@@ -487,7 +487,7 @@ def mapped_fn(
                 positions.shape[0],
             ),
             normalization,
-        )
+        ).to(dtype=positions.dtype)
 
     return mapped_fn
 

diff --git a/tests/beignet/func/test_pair_interactions.py b/tests/beignet/func/test_pair_interactions.py
@@ -16,8 +16,8 @@
 
 PARTICLE_COUNT = 16
 NEIGHBOR_LIST_PARTICLE_COUNT = 16
-POSITION_DTYPE = [torch.float32]  # Example values
-SPATIAL_DIMENSION = [3]
+POSITION_DTYPE = [torch.float32, torch.float64]  # Example values
+SPATIAL_DIMENSION = [2, 3]
 NEIGHBOR_LIST_FORMAT = [
     _NeighborListFormat.DENSE,
     _NeighborListFormat.ORDERED_SPARSE,