code: def _get_data(problem_type: ChemistryProblemType) -> ChemistryData:
      file_path = (
          Path(__file__)
          .parent.joinpath("chemistry_data.zip")
          .absolute()
      )
      with ZipFile(file_path) as zf:
          with zf.open(f"{problem_type.value}.csv") as f:
              df = pd.read_csv(f, index_col=0)
      param_names = sorted(col for col in df.columns if col != "yield")
      return ChemistryData(
          param_names=param_names,
          objective_dict=df.set_index(param_names)["yield"].to_dict(),
      )
  
imports: from __future__ import annotations
  from dataclasses import dataclass
  from enum import Enum
  from functools import lru_cache
  from pathlib import Path
  from typing import Any, Dict, List, Tuple
  from zipfile import ZipFile
  import pandas
  from ax.core.base_trial import BaseTrial
  from ax.core.data import Data
  from ax.core.metric import Metric, MetricFetchE, MetricFetchResult
  from ax.core.types import TParameterization, TParamValue
  from ax.utils.common.result import Err, Ok
  from ax.utils.common.typeutils import not_none
label_desc: Use image node_id 1 for calling a global function with example usage: _get_data(problem_type) and returns: ChemistryData