linting

simplymathematics · Nov 29, 2023 · e362f85 · e362f85
1 parent c3e7f14
commit e362f85
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/deckard/base/model/art_pipeline.py b/deckard/base/model/art_pipeline.py
@@ -64,6 +64,7 @@ def __call__(self):
             tuple(torch_dict.values()),
         ):
             import torch
+
             device_type = "gpu" if torch.cuda.is_available() else "cpu"
             if device_type == "gpu":
                 logger.info("Using GPU")

diff --git a/deckard/base/model/model.py b/deckard/base/model/model.py
@@ -110,7 +110,6 @@ def __init__(self, **kwargs):
         logger.info(f"Initializing model trainer with kwargs {kwargs}")
         self.kwargs = kwargs
 
-
     def __call__(self, data: list, model: object, library=None):
         logger.info(f"Training model {model} with fit params: {self.kwargs}")
         device = str(model.device) if hasattr(model, "device") else "cpu"
@@ -179,6 +178,7 @@ def __call__(self, data: list, model: object, library=None):
                 end_timestamp = time()
             elif "should be the same" in str(e).lower():
                 import torch
+
                 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
                 data[0] = torch.from_numpy(data[0])
                 data[1] = torch.from_numpy(data[1])