From a167e913bd49ad67d95ae11d5646276182833c91 Mon Sep 17 00:00:00 2001 From: lucasmiranda <miranda@biochem.mpg.de> Date: Thu, 14 Sep 2023 11:32:59 +0200 Subject: [PATCH] Updated (S)LEAP input workflow, which now supports .npy, .slp, and .h5 files (testing pending) --- deepof/data.py | 2 +- tests/test_data.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/deepof/data.py b/deepof/data.py index 5b7472ed..6ff80184 100644 --- a/deepof/data.py +++ b/deepof/data.py @@ -141,7 +141,7 @@ class Project: self.trained_path = resource_filename(__name__, "trained_models") # Detect files to load from disk - self.table_format = table_format + self.table_format = table_format.replace(".", "") if self.table_format == "autodetect": ex = [i for i in os.listdir(self.table_path) if not i.startswith(".")][0] self.table_format = ex.split(".")[-1] diff --git a/tests/test_data.py b/tests/test_data.py index 045cd801..c6a07b2a 100644 --- a/tests/test_data.py +++ b/tests/test_data.py @@ -25,7 +25,7 @@ import deepof.utils @settings(max_examples=2, deadline=None) @given( - table_type=st.one_of(st.just(".npy"), st.just(".npy"), st.just(".npy")), + table_type=st.one_of(st.just("h5"), st.just("csv"), st.just("npy")), arena_detection=st.one_of( st.just("circular-autodetect"), st.just("polygonal-autodetect") ), -- GitLab