minor bugfix

Author: Thinh Nguyen

element_array_ephys/__init__.py

@@ -11,7 +11,7 @@ def find_full_path(root_directories, relative_path):
      from provided potential root directories (in the given order)
         :param root_directories: potential root directories
         :param relative_path: the relative path to find the valid root directory
-        :return: root_directory
+        :return: root_directory (pathlib.Path object)
     """
     relative_path = pathlib.Path(relative_path)
 
@@ -36,7 +36,7 @@ def find_root_directory(root_directories, full_path):
     search and return one directory that is the parent of the given path
         :param root_directories: potential root directories
         :param full_path: the relative path to search the root directory
-        :return: full-path
+        :return: full-path (pathlib.Path object)
     """
     full_path = pathlib.Path(full_path)
 
@@ -48,8 +48,9 @@ def find_root_directory(root_directories, full_path):
         root_directories = [root_directories]
 
     try:
-        return next(root_dir for root_dir in root_directories
-                    if full_path.is_relative_to(root_dir))
+        return next(pathlib.Path(root_dir) for root_dir in root_directories
+                    if pathlib.Path(root_dir) in set(full_path.parents))
+
     except StopIteration:
         raise FileNotFoundError('No valid root directory found (from {})'
                                 ' for {}'.format(root_directories, full_path))

element_array_ephys/readers/kilosort.py

@@ -98,10 +98,10 @@ def _stat(self):
                 cluster_file = next(self._kilosort_dir.glob(cluster_pattern))
             except StopIteration:
                 pass
-
-            cluster_file_suffix = cluster_file.suffix
-            assert cluster_file_suffix in ('.csv', '.tsv', '.xlsx')
-            break
+            else:
+                cluster_file_suffix = cluster_file.suffix
+                assert cluster_file_suffix in ('.csv', '.tsv', '.xlsx')
+                break
         else:
             raise FileNotFoundError(
                 'Neither "cluster_groups" nor "cluster_KSLabel" file found!')