Refactor ColumnMapping imports in evidently modules

Author: htahir1

src/zenml/integrations/evidently/column_mapping.py

@@ -15,8 +15,8 @@
 
 from typing import List, Optional, Sequence, Union
 
-from evidently.legacy.pipeline.column_mapping import (
-    ColumnMapping,  # type: ignore[import-untyped]
+from evidently.legacy.pipeline.column_mapping import (  # type: ignore[import-untyped]
+    ColumnMapping,
 )
 from pydantic import BaseModel, ConfigDict, Field
 

src/zenml/integrations/evidently/data_validators/evidently_data_validator.py

@@ -25,11 +25,13 @@
 )
 
 import pandas as pd
-from evidently.legacy.pipeline.column_mapping import (
-    ColumnMapping,  # type: ignore
+from evidently.legacy.pipeline.column_mapping import (  # type: ignore[import-untyped]
+    ColumnMapping,
+)
+from evidently.legacy.report import Report  # type: ignore[import-untyped]
+from evidently.legacy.test_suite import (  # type: ignore[import-untyped]
+    TestSuite,
 )
-from evidently.legacy.report import Report  # type: ignore
-from evidently.legacy.test_suite import TestSuite  # type: ignore
 
 from zenml.data_validators import BaseDataValidator, BaseDataValidatorFlavor
 from zenml.integrations.evidently.flavors.evidently_data_validator_flavor import (

src/zenml/integrations/evidently/steps/evidently_report.py

@@ -108,4 +108,4 @@ def evidently_report_step(
         report_options=report_options or [],
         download_nltk_data=download_nltk_data,
     )
-    return report.json(), HTMLString(report.show(mode="inline").data)
+    return report.json(), HTMLString(report.get_html())

src/zenml/integrations/evidently/steps/evidently_test.py

@@ -104,5 +104,5 @@ def evidently_test_step(
     )
     return (
         test_suite.json(),
-        HTMLString(test_suite.show(mode="inline").data),
+        HTMLString(test_suite.get_html()),
     )