fix(tracker): log_metrics dataframe argument check allows df to be passed

Author: jmahlik

src/smexperiments/tracker.py

@@ -450,7 +450,7 @@ def log_table(self, title=None, values=None, data_frame=None, output_artifact=Tr
                         'Table values should be list. i.e. {"x": [1,2,3]}, instead was ' + str(type(values[key]))
                     )
 
-        if data_frame:
+        if data_frame is not None:
             values = _ArtifactConverter.convert_data_frame_to_values(data_frame)
             fields = _ArtifactConverter.convert_data_frame_to_fields(data_frame)
         else:

tests/unit/test_tracker.py

@@ -588,6 +588,27 @@ def test_log_table(under_test):
     under_test._lineage_artifact_tracker.add_input_artifact("TestTable", "s3uri_value", "etag_value", "Table")
 
 
+def test_log_table_dataframe(under_test):
+
+    dataframe = pd.DataFrame({"x": [1, 2, 3], "y": [4, 5, 6]})
+
+    under_test._artifact_uploader.upload_object_artifact.return_value = ("s3uri_value", "etag_value")
+
+    under_test.log_table(title="TestTable", data_frame=dataframe)
+    expected_data = {
+        "type": "Table",
+        "version": 0,
+        "title": "TestTable",
+        "fields": [{"name": "x", "type": "number"}, {"name": "y", "type": "number"}],
+        "data": {"x": [1, 2, 3], "y": [4, 5, 6]},
+    }
+    under_test._artifact_uploader.upload_object_artifact.assert_called_with(
+        "TestTable", expected_data, file_extension="json"
+    )
+
+    under_test._lineage_artifact_tracker.add_input_artifact("TestTable", "s3uri_value", "etag_value", "Table")
+
+
 def test_log_roc_curve(under_test):
     y_true = [0, 0, 1, 1]
     y_scores = [0.1, 0.4, 0.35, 0.8]