From b85de694d394e829e9b6939d894f1200e67f069b Mon Sep 17 00:00:00 2001 From: polyaxon-ci Date: Thu, 28 Sep 2023 12:19:15 +0200 Subject: [PATCH] Update clipped imports --- traceml/traceml/events/schemas.py | 4 ++-- traceml/traceml/logging/schemas.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/traceml/traceml/events/schemas.py b/traceml/traceml/events/schemas.py index 1500c93f1..07eb12664 100644 --- 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 index eb509505d..a226e516c 100644 --- 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,