Remove unnecessary dependency on structure test_utils

PiperOrigin-RevId: 765132514
Change-Id: Ia1acdf55267083741c294864792f25009f5f4542
This commit is contained in:
Augustin Zidek
2025-05-30 04:24:50 -07:00
committed by Copybara-Service
parent 565f286892
commit 6a34fea9a2
2 changed files with 2 additions and 4 deletions

View File

@ -31,7 +31,6 @@ from alphafold3.constants import chemical_components
from alphafold3.data import featurisation
from alphafold3.data import pipeline
from alphafold3.model.atom_layout import atom_layout
from alphafold3.structure import test_utils
import jax
import numpy as np
@ -98,7 +97,7 @@ def _generate_diff(actual: str, expected: str) -> str:
)
class DataPipelineTest(test_utils.StructureTestCase):
class DataPipelineTest(parameterized.TestCase):
"""Test AlphaFold 3 inference."""
def setUp(self):

View File

@ -28,7 +28,6 @@ from alphafold3.common import resources
from alphafold3.common.testing import data as testing_data
from alphafold3.data import pipeline
from alphafold3.model.scoring import alignment
from alphafold3.structure import test_utils
import jax
import numpy as np
@ -64,7 +63,7 @@ def _generate_diff(actual: str, expected: str) -> str:
)
class InferenceTest(test_utils.StructureTestCase):
class InferenceTest(parameterized.TestCase):
"""Test AlphaFold 3 inference."""
def setUp(self):