From e1d28157d0cbae0736cd54bb4949341aa3ba00fc Mon Sep 17 00:00:00 2001 From: Yee Hing Tong Date: Fri, 24 Jan 2025 17:14:14 -0800 Subject: [PATCH] Add back import and format imports better (#3089) Signed-off-by: Yee Hing Tong --- tests/flytekit/unit/core/test_dataclass.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/flytekit/unit/core/test_dataclass.py b/tests/flytekit/unit/core/test_dataclass.py index 80f2f56b56..d49031e390 100644 --- a/tests/flytekit/unit/core/test_dataclass.py +++ b/tests/flytekit/unit/core/test_dataclass.py @@ -1,11 +1,12 @@ import pytest -from enum import Enum import mock -from pathlib import Path -from mashumaro.mixins.json import DataClassJSONMixin import os import sys import tempfile +from enum import Enum +from pathlib import Path +from dataclasses_json import DataClassJsonMixin +from mashumaro.mixins.json import DataClassJSONMixin from dataclasses import dataclass, fields, field from typing import List, Dict, Optional, Union, Any from typing_extensions import Annotated