From 6b61e754ba7d21267d5e2d83c4d89893b10f3714 Mon Sep 17 00:00:00 2001
From: PhiSpel <62760840+PhiSpel@users.noreply.github.com>
Date: Wed, 11 Sep 2024 12:29:58 +0200
Subject: [PATCH] Update test_einsum.py (#249)

* Update test_einsum.py

* Update test_einsum.py
---
 tests/flow/test_einsum.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/flow/test_einsum.py b/tests/flow/test_einsum.py
index 8aa85f30..c93c77fc 100644
--- 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)