Fix pandas merge being silly

UCSD-E4E · Jul 5, 2023 · 43df354 · 43df354
1 parent 99890c1
commit 43df354
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/classification/dataset.py b/classification/dataset.py
@@ -126,14 +126,14 @@ def serialize_data(self):
                        right_on="IN FILE").dropna()
 
         print("fixed size:", self.samples.shape)
-
-        self.samples["original_file_path"] = self.samples[self.config.file_path_col]
 
         if "files" in self.samples.columns:
             self.samples[self.config.file_path_col] = self.samples["files"].copy()
         if "files_y" in self.samples.columns:
             self.samples[self.config.file_path_col] = self.samples["files_y"].copy()
 
+        self.samples["original_file_path"] = self.samples[self.config.file_path_col]
+
         self.formatted_csv_file = ".".join(self.csv_file.split(".")[:-1]) + "formatted.csv"
         self.samples.to_csv(self.formatted_csv_file)