Update clipped imports

polyaxon · Sep 28, 2023 · b85de69 · b85de69
1 parent 5b1c74f
commit b85de69
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/traceml/traceml/events/schemas.py b/traceml/traceml/events/schemas.py
@@ -7,10 +7,10 @@
 from clipped.compact.pydantic import StrictStr, root_validator
 from clipped.config.parser import ConfigParser
 from clipped.config.schema import skip_partial
-from clipped.utils.csv import validate_csv
 from clipped.utils.dates import parse_datetime
 from clipped.utils.enums import PEnum
 from clipped.utils.np import sanitize_np_types
+from clipped.utils.strings import validate_file_or_buffer
 from clipped.utils.tz import now
 
 from polyaxon._schemas.base import BaseSchemaModel
@@ -431,7 +431,7 @@ def read(
         import pandas as pd
 
         if isinstance(data, str):
-            data = validate_csv(data)
+            data = validate_file_or_buffer(data)
             error = None
             if V1ArtifactKind.is_jsonl_file_event(kind):
                 try:

diff --git a/traceml/traceml/logging/schemas.py b/traceml/traceml/logging/schemas.py
@@ -3,9 +3,9 @@
 from typing import List, Optional, Text
 
 from clipped.compact.pydantic import StrictStr
-from clipped.utils.csv import validate_csv
 from clipped.utils.dates import parse_datetime
 from clipped.utils.json import orjson_dumps, orjson_loads
+from clipped.utils.strings import validate_file_or_buffer
 from clipped.utils.tz import now
 
 from polyaxon._schemas.base import BaseSchemaModel
@@ -105,7 +105,7 @@ def read_csv(cls, data: str, parse_dates: bool = True) -> "V1Logs":
         import numpy as np
         import pandas as pd
 
-        csv = validate_csv(data)
+        csv = validate_file_or_buffer(data)
         if parse_dates:
             df = pd.read_csv(
                 csv,