Merge branch 'nuo_dev' of https://github.com/LearningCircuitsLab/lecilab-behavior-analysis into nuo_dev

Author: tudou2333333

lecilab_behavior_analysis/df_transforms.py

@@ -121,7 +121,9 @@ def add_port_where_animal_comes_from(df_in: pd.DataFrame) -> pd.DataFrame:
             # equal to the original dataframe
             df.loc[df_mouse_session.index, 'roa_choice'] = series_to_append
             # add also the port where the animal comes from
-            df.loc[df_mouse_session.index, 'previous_port_before_stimulus'] = last_choice
+            last_choice_series = pd.Series(last_choice, index=df_mouse_session.index)
+            df.loc[df_mouse_session.index, 'previous_port_before_stimulus'] = last_choice_series.astype("string")
+
 
     return df
 

lecilab_behavior_analysis/utils.py

@@ -317,7 +317,7 @@ def get_outpath():
         "lorena-ThinkPad-E550": "/home/emma/Desktop/EloiJacomet/data",
         "tectum": "/mnt/c/Users/HMARTINEZ/LeCiLab/data/behavioral_data",
         "tudou": "/home/kudongdong/data/LeciLab/behavioral_data",
-        "nuo-rostower": "/home/kudongdong/Documents/data/LeciLab/behavioral_data",
+        "nuo-rostower": "/home/nuo/Documents/data/LeciLab/behavioral_data",
         "minibaps": "/archive/training_village",
     }
     return paths.get(hostname, "default/path")
@@ -408,6 +408,12 @@ def rsync_specific_file(
     """
     This method syncs the session data from the server to the local machine.
     """
+    if not os.path.isdir(local_path):
+        try:
+            os.makedirs(local_path)  # Creates the directory (and parents if needed)
+            print(f"Directory '{local_path}' created.")
+        except OSError as e:
+            print(f"Error creating directory '{local_path}': {e}")
     remote_path = f"{credentials['username']}@{credentials['host']}:{file_path}"
     rsync_command = f"rsync -avz {remote_path} {local_path}"
     result = subprocess.run(rsync_command, shell=True)
@@ -1109,7 +1115,10 @@ def get_previous_row_index(df: pd.DataFrame, current_index) -> int:
         raise IndexError("No previous row exists for the given index.")
 
 
-def transform_side_choice_to_numeric(side: str) -> Union[int, float]:
+def transform_side_choice_to_numeric(side) -> Union[int, float]:
+    if pd.isna(side):
+        return np.nan
+
     match side:
         case "left":
             return 1