import leakpro

aidotse · Sep 6, 2024 · 6873886 · 6873886
1 parent b976fba
commit 6873886
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/leakpro/tests/conftest.py b/leakpro/tests/conftest.py
@@ -8,6 +8,7 @@
 import pytest
 from dotmap import DotMap
 
+
 from input_handler.image_utils import setup_image_test
 from input_handler.cifar10_input_handler import Cifar10InputHandler
 from constants import STORAGE_PATH

diff --git a/leakpro/tests/input_handler/cifar10_input_handler.py b/leakpro/tests/input_handler/cifar10_input_handler.py
@@ -7,6 +7,7 @@
 from torch.utils.data import DataLoader
 from tqdm import tqdm
 
+import leakpro
 from leakpro.import_helper import Self
 from leakpro.user_inputs.abstract_input_handler import AbstractInputHandler