From c5c07a40c598cd6ed9b4479c2a14064286ad6e9e Mon Sep 17 00:00:00 2001 From: rockerBOO Date: Tue, 25 Mar 2025 18:41:58 -0400 Subject: [PATCH] Move test util to library --- {tests => library}/test_util.py | 0 tests/__init__.py | 0 tests/library/test_lora_util.py | 2 +- tests/networks/test_lora_flux.py | 2 +- 4 files changed, 2 insertions(+), 2 deletions(-) rename {tests => library}/test_util.py (100%) delete mode 100644 tests/__init__.py diff --git a/tests/test_util.py b/library/test_util.py similarity index 100% rename from tests/test_util.py rename to library/test_util.py diff --git a/tests/__init__.py b/tests/__init__.py deleted file mode 100644 index e69de29b..00000000 diff --git a/tests/library/test_lora_util.py b/tests/library/test_lora_util.py index df5243ba..60273464 100644 --- a/tests/library/test_lora_util.py +++ b/tests/library/test_lora_util.py @@ -1,7 +1,7 @@ import torch import pytest from library.lora_util import initialize_pissa -from ..test_util import generate_synthetic_weights +from library.test_util import generate_synthetic_weights def test_initialize_pissa_basic(): diff --git a/tests/networks/test_lora_flux.py b/tests/networks/test_lora_flux.py index 86ccb376..5f141e4f 100644 --- a/tests/networks/test_lora_flux.py +++ b/tests/networks/test_lora_flux.py @@ -2,7 +2,7 @@ import pytest import torch import torch.nn as nn from networks.lora_flux import LoRAModule, LoRANetwork, create_network -from ..test_util import generate_synthetic_weights +from library.test_util import generate_synthetic_weights from unittest.mock import MagicMock