Update test_einsum.py (#249)

* Update test_einsum.py * Update test_einsum.py
lettucecfd · Sep 11, 2024 · 6b61e75 · 6b61e75
1 parent d00c4c0
commit 6b61e75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/flow/test_einsum.py b/tests/flow/test_einsum.py
@@ -170,7 +170,7 @@ def source_term(self, u):
 
 @pytest.mark.parametrize("fix_dim", [1, 2, 3])
 def test_einsum(fix_dim):
-    context = Context()
+    context = Context(dtype=torch.float64)
 
     flow = EinsumFlow(context, [16] * fix_dim)
     flow.f = torch.rand_like(flow.f)