Revert "fix cosmo"

This reverts commit 8f4d844.

Author: gmertes

src/anemoi/inference/lazy.py

@@ -34,8 +34,6 @@ def __getattr__(self, name: str):
 # with type checking and autocompletion: `from anemoi.inference.lazy import torch`
 # note: when used in a type hint, use quotes, e.g. "torch.Tensor" instead of torch.Tensor to avoid triggering the import
 if TYPE_CHECKING:
-    import earthkit.data as ekd
     import torch
 else:
-    ekd = LazyModule("earthkit.data")
     torch = LazyModule("torch")

tests/unit/test_templates.py

@@ -2,11 +2,11 @@
 
 import yaml
 from anemoi.utils.testing import GetTestData
+from earthkit.data.readers.grib.codes import GribField
 from pytest_mock import MockerFixture
 
 from anemoi.inference.checkpoint import Checkpoint
 from anemoi.inference.grib.templates.manager import TemplateManager
-from anemoi.inference.lazy import ekd
 from anemoi.inference.testing import fake_checkpoints
 
 
@@ -19,7 +19,7 @@ def test_manager_builtin(mocker: MockerFixture):
     manager = TemplateManager(owner)
 
     template = manager.template("2t", state={}, typed_variables=c.typed_variables)
-    assert isinstance(template, ekd.Field)
+    assert isinstance(template, GribField)
     assert template.metadata("param") == "lsm"  # lsm is used as the builtin template for surface fields
 
 
@@ -36,7 +36,7 @@ def test_manager_file(mocker: MockerFixture, get_test_data: GetTestData):
     manager = TemplateManager(owner, templates=config)
 
     template = manager.template("2t", state={}, typed_variables=c.typed_variables)
-    assert isinstance(template, ekd.Field)
+    assert isinstance(template, GribField)
     assert template.metadata("param") == "10u"  # first field in the file
 
 
@@ -62,7 +62,7 @@ def test_manager_samples(mocker: MockerFixture, get_test_data: GetTestData, tmp_
     manager = TemplateManager(owner, templates=config)
 
     template = manager.template("2t", state={}, typed_variables=c.typed_variables)
-    assert isinstance(template, ekd.Field)
+    assert isinstance(template, GribField)
     assert template.metadata("param") == "10u"  # first field in the file