bug parenthesis erased

supernnova · Apr 6, 2020 · 00ff200 · 00ff200
1 parent 3211bb9
commit 00ff200
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/supernnova/utils/data_utils.py b/supernnova/utils/data_utils.py
@@ -138,6 +138,7 @@ def tag_type(df, settings, type_column="TYPE"):
 
     df[f"target_{len(settings.sntypes)}classes"] = df[type_column].apply(
         lambda x: classes_to_use[x]
+        )
 
     return df