Fix copybara OSS export test failures

dpsynth/eval/correlation_computation.py

@@ -15,6 +15,7 @@
 """Computes two-way attribute correlations."""
 
 from collections.abc import Iterable
+import dataclasses
 import itertools
 from typing import Any
 from dpsynth.eval import types

dpsynth/eval/one_way_distribution_computation.py

@@ -15,6 +15,7 @@
 """Computes one-way marginal distances."""
 
 from collections.abc import Iterable
+import dataclasses
 from typing import Any
 from dpsynth.eval import types
 from dpsynth.pipeline_transformations import diagnostic_info

dpsynth/pipeline_transformations/diagnostic_info.py

@@ -15,6 +15,10 @@
 """Module for updating diagnostic information."""
 
 import copy
+
+from dpsynth.pipeline_transformations import types  # pylint: disable=g-bad-import-order
+import pipeline_dp  # pylint: disable=g-bad-import-order
+
 from dataclasses import dataclass, field
 from typing import Optional
 

pyproject.toml

@@ -51,3 +51,6 @@ version = {attr = "dpsynth.__version__"}
 
 [tool.setuptools.packages.find]
 namespaces = false
+
+[tool.pytest.ini_options]
+addopts = "--import-mode=importlib"