fix: tiny bug fix to avoid KeyError (#2264)

microsoft · Aug 12, 2024 · eb309fe · eb309fe
1 parent 72f99c8
commit eb309fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/main/python/synapse/ml/core/logging/SynapseMLLogger.py b/core/src/main/python/synapse/ml/core/logging/SynapseMLLogger.py
@@ -293,7 +293,7 @@ def log_fit():
 
     @classmethod
     def get_column_number(cls, args, kwargs):
-        if kwargs and kwargs["df"] and isinstance(kwargs["df"], DataFrame):
+        if kwargs and kwargs.get("df") and isinstance(kwargs["df"], DataFrame):
             return len(kwargs["df"].columns)
         elif args and len(args) > 0 and isinstance(args[0], DataFrame):
             return len(args[0].columns)