From 98c4b7753b2cf9b467bdda0c94569b5b910262fa Mon Sep 17 00:00:00 2001 From: Stephan Auerhahn Date: Sat, 12 Aug 2023 07:16:02 +0000 Subject: [PATCH] cleanup imports in test --- tests/inference/test_modelmanager.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/tests/inference/test_modelmanager.py b/tests/inference/test_modelmanager.py index fa96d70..bb1ab0e 100644 --- a/tests/inference/test_modelmanager.py +++ b/tests/inference/test_modelmanager.py @@ -1,15 +1,8 @@ -import numpy -from PIL import Image import pytest -from pytest import fixture import torch -from typing import Tuple, Optional -from sgm.inference.api import ( - model_specs, - SamplingParams, - SamplingPipeline, - Sampler, +from sgm.inference.api import ( + SamplingPipeline, ModelArchitecture, ) import sgm.inference.helpers as helpers