fixed if statement

bhklab · Jun 21, 2022 · c6a5f4d · c6a5f4d
1 parent 13f90fd
commit c6a5f4d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/autotest.py b/examples/autotest.py
@@ -373,7 +373,7 @@ def save_data(self):
         folder_renames = {}
         for col in self.output_df.columns:
             if col.startswith("folder"):
-                self.output_df[col] = self.output_df[col].apply(lambda x: x if isinstance(x, str) else pathlib.Path(x).as_posix().split(self.input_directory)[1][1:]) # rel path, exclude the slash at the beginning
+                self.output_df[col] = self.output_df[col].apply(lambda x: x if not isinstance(x, str) else pathlib.Path(x).as_posix().split(self.input_directory)[1][1:]) # rel path, exclude the slash at the beginning
                 folder_renames[col] = f"input_{col}"
         self.output_df.rename(columns=folder_renames, inplace=True) #append input_ to the column name
         self.output_df.to_csv(self.output_df_path)

diff --git a/imgtools/autopipeline.py b/imgtools/autopipeline.py
@@ -379,7 +379,7 @@ def save_data(self):
         folder_renames = {}
         for col in self.output_df.columns:
             if col.startswith("folder"):
-                self.output_df[col] = self.output_df[col].apply(lambda x: x if isinstance(x, str) else pathlib.Path(x).as_posix().split(self.input_directory)[1][1:]) # rel path, exclude the slash at the beginning
+                self.output_df[col] = self.output_df[col].apply(lambda x: x if not isinstance(x, str) else pathlib.Path(x).as_posix().split(self.input_directory)[1][1:]) # rel path, exclude the slash at the beginning
                 folder_renames[col] = f"input_{col}"
         self.output_df.rename(columns=folder_renames, inplace=True) #append input_ to the column name
         self.output_df.to_csv(self.output_df_path)