diff --git a/src/aarecommon/config/beamline.py b/src/aarecommon/config/beamline.py index 8ca01e8..62ec360 100644 --- a/src/aarecommon/config/beamline.py +++ b/src/aarecommon/config/beamline.py @@ -1,6 +1,6 @@ import os from pathlib import Path -from typing import Any, Dict +from typing import Any import yaml @@ -12,7 +12,7 @@ class BeamlineYAMLConfig: self.beamline: MXBeamline = mx_beamline() self.config = self._load_config() - def _load_config(self) -> Dict[str, Any]: + def _load_config(self) -> dict[str, Any]: config_dir = Path(__file__).parent / "beamline_configs" yaml_file = config_dir / f"{self.beamline.value.lower()}.yaml"