fix missing arg

SMLC-NYSBC · Apr 19, 2023 · cfff75f · cfff75f
1 parent 374e519
commit cfff75f
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 13 deletions.
diff --git a/.licenserc.json b/.licenserc.json
@@ -2,22 +2,24 @@
   "**/*.py": "#  MIT License 2021 - 2023                                            #",
 
   "ignore": [
-    "tardis/__init__.py",
+    "**/__init__.py",
 
-    "tardis/utils/__init__.py",
+    "tardis_pytorch/__init__.py",
 
-    "tardis/spindletorch/__init__.py",
-    "tardis/spindletorch/data_processing/__init__.py",
-    "tardis/spindletorch/datasets/__init__.py",
-    "tardis/spindletorch/model/__init__.py",
-    "tardis/spindletorch/utils/__init__.py",
+    "tardis_pytorch/utils/__init__.py",
 
-    "tardis/dist_pytorch/__init__.py",
-    "tardis/dist_pytorch/datasets/__init__.py",
-    "tardis/dist_pytorch/model/__init__.py",
-    "tardis/dist_pytorch/utils/__init__.py",
+    "tardis_pytorch/spindletorch/__init__.py",
+    "tardis_pytorch/spindletorch/data_processing/__init__.py",
+    "tardis_pytorch/spindletorch/datasets/__init__.py",
+    "tardis_pytorch/spindletorch/model/__init__.py",
+    "tardis_pytorch/spindletorch/utils/__init__.py",
 
-    "tardis/benchmarks/__init__.py",
+    "tardis_pytorch/dist_pytorch/__init__.py",
+    "tardis_pytorch/dist_pytorch/datasets/__init__.py",
+    "tardis_pytorch/dist_pytorch/model/__init__.py",
+    "tardis_pytorch/dist_pytorch/utils/__init__.py",
+
+    "tardis_pytorch/benchmarks/__init__.py",
 
     "docs/"
   ]

diff --git a/tardis_pytorch/tardis.py b/tardis_pytorch/tardis.py
@@ -1,3 +1,12 @@
+#######################################################################
+#  TARDIS - Transformer And Rapid Dimensionless Instance Segmentation #
+#                                                                     #
+#  New York Structural Biology Center                                 #
+#  Simons Machine Learning Center                                     #
+#                                                                     #
+#  Robert Kiewisz, Tristan Bepler                                     #
+#  MIT License 2021 - 2023                                            #
+#######################################################################
 import click
 from tardis_pytorch._version import version
 from tardis_pytorch.utils.logo import TardisLogo

diff --git a/tardis_pytorch/utils/predictor.py b/tardis_pytorch/utils/predictor.py
@@ -819,7 +819,7 @@ def __call__(self, *args, **kwargs):
 
                 if self.predict in ["Filament", "Microtubule"]:
                     np.savetxt(
-                        join(self.am_output, f"{i[:-self.in_format]}_Segments.csv"),
+                        join(self.am_output, f"{i[:-self.in_format]}_Segments_filter.csv"),
                         self.filter_splines(segments=self.segments),
                         delimiter=",",
                     )