From 012981b9dface2c82eb6a365be1d3966e7239907 Mon Sep 17 00:00:00 2001 From: Kevin Santana Date: Sun, 20 Oct 2024 16:08:57 -0700 Subject: [PATCH] typing fix --- clouddrift/adapters/ibtracs.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/clouddrift/adapters/ibtracs.py b/clouddrift/adapters/ibtracs.py index 86574b08..b45975bd 100644 --- a/clouddrift/adapters/ibtracs.py +++ b/clouddrift/adapters/ibtracs.py @@ -1,6 +1,6 @@ import os import tempfile -from typing import Literal, TypeAlias +from typing import Hashable, Literal, TypeAlias import numpy as np import xarray as xr @@ -90,11 +90,12 @@ def to_raggedarray( ds = xr.open_dataset(dst_path, engine="netcdf4") ds = ds.rename_dims({"date_time": "obs"}) - vars = list() - vars.extend(ds.variables) - vars -= ds.coords.keys() + vars = list[Hashable]() + vars.extend(ds.variables.keys()) + for coord in ds.coords.keys(): + vars.remove(coord) dtypes = {v: ds[v].dtype for v in vars} - dtypes.update({"numobs": np.int64}) + dtypes.update({"numobs": np.dtype("int64")}) ds = ds.astype(dtypes) data_vars = list()