From 3b6657a8f370c8f1f80ac602b60e033f30d02e3c Mon Sep 17 00:00:00 2001 From: Chris McBride Date: Thu, 17 Oct 2024 16:53:36 -0500 Subject: [PATCH] isorts on _legacy --- tests/_legacy/backends/run_torch.py | 2 +- tests/_legacy/test_model.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/_legacy/backends/run_torch.py b/tests/_legacy/backends/run_torch.py index 48fb9dcca..1071e740e 100644 --- a/tests/_legacy/backends/run_torch.py +++ b/tests/_legacy/backends/run_torch.py @@ -25,12 +25,12 @@ # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. import io +import typing as t import numpy as np import torch import torch.nn as nn import torch.nn.functional as F -import typing as t from smartredis import Client diff --git a/tests/_legacy/test_model.py b/tests/_legacy/test_model.py index f57d59fc0..7fc43ae94 100644 --- a/tests/_legacy/test_model.py +++ b/tests/_legacy/test_model.py @@ -24,21 +24,20 @@ # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +import typing as t from uuid import uuid4 import numpy as np import pytest -import typing as t from smartsim import Experiment from smartsim._core.control.manifest import LaunchedManifestBuilder from smartsim._core.launcher.step import SbatchStep, SrunStep - from smartsim.builders import Ensemble from smartsim.entity import Application - from smartsim.error import EntityExistsError, SSUnsupportedError from smartsim.settings import RunSettings, SbatchSettings, SrunSettings + # from smartsim.settings.mpiSettings import _BaseMPISettings = t.Any