From fb1270d713aff512a92cd1f4824cc849fcc4c3b2 Mon Sep 17 00:00:00 2001
From: generatedunixname89002005279527
 <generatedunixname89002005279527@meta.com>
Date: Tue, 4 Jun 2024 11:39:33 -0700
Subject: [PATCH] Remove unused imports from ax/ (#2501)

Summary: Pull Request resolved: https://github.com/facebook/Ax/pull/2501

Reviewed By: mpolson64

Differential Revision: D58010705

fbshipit-source-id: bfe6d0897044f5f448fbc9a4cd783c5a3449944c
---
 ax/__init__.py      | 4 +---
 ax/core/__init__.py | 3 +--
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/ax/__init__.py b/ax/__init__.py
index 5d262606c41..d1f2e9e6f65 100644
--- a/ax/__init__.py
+++ b/ax/__init__.py
@@ -37,9 +37,7 @@
 from ax.storage import json_load, json_save
 
 try:
-    # Marking this as a manual import to avoid autodeps complaints
-    # due to imports from non-existent file.
-    from ax.version import version as __version__  # @manual  # pyre-ignore[21]
+    pass
 except Exception:  # pragma: no cover
     __version__ = "Unknown"
 
diff --git a/ax/core/__init__.py b/ax/core/__init__.py
index db93aa44ff7..9b6fa1243a9 100644
--- a/ax/core/__init__.py
+++ b/ax/core/__init__.py
@@ -14,7 +14,7 @@
 from ax.core.generator_run import GeneratorRun
 from ax.core.metric import Metric
 from ax.core.objective import MultiObjective, Objective
-from ax.core.observation import Observation, ObservationData, ObservationFeatures
+from ax.core.observation import ObservationFeatures
 from ax.core.optimization_config import (
     MultiObjectiveOptimizationConfig,
     OptimizationConfig,
@@ -41,7 +41,6 @@
 from ax.core.runner import Runner
 from ax.core.search_space import SearchSpace
 from ax.core.trial import Trial
-from ax.core.types import TParameterization
 
 
 __all__ = [