Fix linting and mypy issue

equinor · Oct 27, 2023 · 4aa83c5 · 4aa83c5
1 parent 7b0ff05
commit 4aa83c5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/subscript/fmuobs/fmuobs.py b/src/subscript/fmuobs/fmuobs.py
@@ -348,7 +348,7 @@ def fmuobs(
 
     # Trigger warning if user specify ERROR_MODE != ABS
     # in BLOCK_OBSERVATION and SUMMARY_OBSERVATION
-    if "ERROR_MODE" in dframe.columns:
+    if isinstance(dframe, pd.DataFrame) and "ERROR_MODE" in dframe.columns:
         error_mode = list(
             dframe[
                 (dframe["CLASS"].isin(["BLOCK_OBSERVATION", "SUMMARY_OBSERVATION"]))
@@ -359,7 +359,8 @@ def fmuobs(
         )
         if len(error_mode) > 0:
             logger.warn(
-                f"Unsupported ERROR_MODE : {', '.join(error_mode)}. Please verify the output file"
+                f"Unsupported ERROR_MODE : {', '.join(error_mode)}. "
+                "Please verify the output file"
             )
 
     dump_results(dframe, csv, yml, resinsight, ertobs)