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,