Adapt to upstream typing changes

lsst · Jun 20, 2024 · d3a9b89 · d3a9b89
1 parent 7f58b53
commit d3a9b89
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/lsst/ci/middleware/output_repo_tests.py b/python/lsst/ci/middleware/output_repo_tests.py
@@ -134,6 +134,8 @@ def check_objects(self, test_case: unittest.TestCase) -> None:
             test_case.assertIsNone(tract_dataset.converted_from)
             assert tract_dataset.quantum is not None
             for patch_dataset_as_input in tract_dataset.quantum.inputs["inputCatalogs"]:
+                if not isinstance(patch_dataset_as_input, MockDataset):
+                    continue
                 patch = cast(int, patch_dataset_as_input.data_id["patch"])
                 patch_ref = patch_refs[tract, patch]
                 # We pre-registered this dataset type with ArrowTable as its
@@ -171,7 +173,7 @@ def check_coadds(self, test_case: unittest.TestCase) -> None:
             if (expected := self.expected[key]) is not None:
                 test_case.assertEqual(
                     {
-                        input_dataset.data_id["visit"]
+                        cast(MockDataset, input_dataset).data_id["visit"]
                         for input_dataset in cast(MockDatasetQuantum, dataset.quantum).inputs["inputWarps"]
                     },
                     expected,