Fix

openclimatefix · Jun 9, 2024 · f5dc5d5 · f5dc5d5
1 parent 6a944ad
commit f5dc5d5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/graph_weather/models/gencast/README.md b/graph_weather/models/gencast/README.md
@@ -0,0 +1,2 @@
+# GenCast
+
diff --git a/graph_weather/models/gencast/graph/graph_builder.py b/graph_weather/models/gencast/graph/graph_builder.py
@@ -71,7 +71,7 @@ def __init__(
         grid_lat: np.ndarray,
         splits: int = 5,
         num_hops: int = 0,
-        device: str = "cpu",
+        device: torch.device = torch.device("cpu"),
     ):
         """Initialize the GraphBuilder object.