version bump

Kapoorlabs-CAPED · Feb 5, 2024 · 6ddd8ea · 6ddd8ea
1 parent 7fe69e0
commit 6ddd8ea
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/oneat/NEATUtils/TrainDataMaker.py b/src/oneat/NEATUtils/TrainDataMaker.py
@@ -10,11 +10,13 @@ class TrainDataMaker:
     def __init__(
         self,
         source_dir,
+        save_dir=None,
         create_2dt=False,
         class_names=["Normal", "Mitosis", "Apoptosis"],
     ):
 
         self.source_dir = source_dir
+        self.save_dir = save_dir if save_dir else source_dir
         self.create_2dt = create_2dt
         self.class_names = class_names
         self.acceptable_formats = [".tif", ".TIFF", ".TIF", ".png"]
@@ -45,7 +47,7 @@ def _click_maker(self):
                 self.viewer,
                 imageidbox.currentText(),
                 os.path.basename(os.path.splitext(imageidbox.currentText())[0]),
-                self.source_dir,
+                self.save_dir,
                 save=False,
                 newimage=True,
                 create_2dt=self.create_2dt,
@@ -58,7 +60,7 @@ def _click_maker(self):
                 self.viewer,
                 imageidbox.currentText(),
                 os.path.basename(os.path.splitext(imageidbox.currentText())[0]),
-                self.source_dir,
+                self.save_dir,
                 save=True,
                 newimage=False,
                 create_2dt=self.create_2dt,

diff --git a/src/oneat/_version.py b/src/oneat/_version.py
@@ -1,2 +1,2 @@
-__version__ = version = "6.7.7"
-__version_tuple__ = version_tuple = (6, 7, 7)
+__version__ = version = "6.7.8"
+__version_tuple__ = version_tuple = (6, 7, 8)