continued fixes

lehner · Jan 1, 2025 · 79fb454 · 79fb454
1 parent d81c950
commit 79fb454
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lib/gpt/qcd/sparse_propagator.py b/lib/gpt/qcd/sparse_propagator.py
@@ -261,7 +261,7 @@ def flavor(root, *cache_param):
 
     # TODO: if root is a list, return a list of flavors that have
     # a uniform sink sparse domain
-    tag, prec = root.split(".")
+    tag, prec = root.rsplit(".", maxsplit=1)
 
     ttag = f"{tag}.{prec}"
     if ttag in prop:

diff --git a/tests/qcd/sparse_propagator.py b/tests/qcd/sparse_propagator.py
@@ -9,17 +9,17 @@
 
 # workdir
 if "WORK_DIR" in os.environ:
-    work_dir = os.environ["WORK_DIR"] + "/"
+    work_dir = os.environ["WORK_DIR"]
 else:
-    work_dir = ""
+    work_dir = "."
 
 # request test files
 files = ["30_combined.zip"]
 for f in files:
-    g.repository.load(f"{work_dir}{f}", f"gpt://tests/qcd/propagators/{f}")
+    g.repository.load(f"{work_dir}/{f}", f"gpt://tests/qcd/propagators/{f}")
 
 cache_params = (16, 4)
-quark = g.qcd.sparse_propagator.flavor(f"{work_dir}30_combined/light.e", *cache_params)
+quark = g.qcd.sparse_propagator.flavor(f"{work_dir}/30_combined/light.e", *cache_params)
 
 coordinates = quark.sink_domain.coordinates
 nsrc = quark.source_domain.sampled_sites