Skip to content

Commit

Permalink
Fixed mypy errors :rage1:
Browse files Browse the repository at this point in the history
  • Loading branch information
mpluess committed Oct 2, 2024
1 parent 5e368e6 commit e310550
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
4 changes: 3 additions & 1 deletion karabo/simulation/interferometer.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,9 @@
from karabo.util.dask import DaskHandler
from karabo.util.file_handler import FileHandler
from karabo.util.gpu_util import is_cuda_available
from karabo.util.ska_sdp_datamodels.visibility.vis_io_ms import export_visibility_to_ms
from karabo.util.ska_sdp_datamodels.visibility.vis_io_ms import ( # type: ignore[attr-defined] # noqa: E501
export_visibility_to_ms,
)


class CorrelationType(enum.Enum):
Expand Down
6 changes: 2 additions & 4 deletions karabo/simulation/sample_simulation.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,9 +50,7 @@ def run_sample_simulation(

if verbose:
print("Setting Up Telescope")
telescope = Telescope.constructor(
"ASKAP", version=None, backend=simulator_backend
) # type: ignore[call-overload]
telescope = Telescope.constructor("ASKAP", version=None, backend=simulator_backend)

if verbose:
print("Setting Up Observation")
Expand All @@ -75,7 +73,7 @@ def run_sample_simulation(
observation,
backend=simulator_backend,
visibility_format=visibility_format,
) # type: ignore[call-overload]
)

# In case run_simulation returns a list of vis (allowed by type hint)
if isinstance(visibility, list):
Expand Down

0 comments on commit e310550

Please sign in to comment.