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