diff --git a/dfm_tools/xugrid_helpers.py b/dfm_tools/xugrid_helpers.py index f05276c1..507972de 100644 --- a/dfm_tools/xugrid_helpers.py +++ b/dfm_tools/xugrid_helpers.py @@ -33,10 +33,7 @@ def get_vertical_dimensions(uds): #TODO: maybe add layer_dimension and interface processing MB_02_0*_map.nc >> found layer/interface dimensions in file: mesh2d_nLayers mesh2d_nInterfaces """ - - if not isinstance(uds,(xr.Dataset,xu.UgridDataset)): - raise TypeError(f'uds should be of type xr.Dataset or xu.UgridDataset, but is {type(uds)}') - + if not hasattr(uds,'grid'): #early return in case of e.g. hisfile return None, None diff --git a/tests/test_xugrid_helpers.py b/tests/test_xugrid_helpers.py index aede2206..78ac27a5 100644 --- a/tests/test_xugrid_helpers.py +++ b/tests/test_xugrid_helpers.py @@ -197,11 +197,6 @@ def test_get_vertical_dimensions(): assert dimn_layer is None assert dimn_interface is None - # incorrect type - with pytest.raises(TypeError) as e: - get_vertical_dimensions(1) - assert 'uds should be of type xr.Dataset or xu.UgridDataset' in str(e.value) - @pytest.mark.systemtest def test_uda_edges_to_faces_interfaces_to_centers():