Accept single file feature generation target

wonkoderverstaendige · Dec 11, 2019 · 73aabbd · 73aabbd
1 parent 983da95
commit 73aabbd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dataman/features/features.py b/dataman/features/features.py
@@ -208,7 +208,11 @@ def main(args):
     cli_args = parser.parse_args(args)
 
     matpath = Path(cli_args.target).resolve()
-    matfiles = sorted(list(map(Path.resolve, matpath.glob('tetrode??.mat'))))
+    if matpath.is_file():
+        matfiles = [matpath]
+    else:
+        matfiles = sorted(list(map(Path.resolve, matpath.glob('tetrode??.mat'))))
+
     logger.debug([mf.name for mf in matfiles])
     logger.info('Found {} waveform files'.format(len(matfiles)))