fix typo

a-r-j · Feb 9, 2024 · c7f8aa4 · c7f8aa4
1 parent 86aba6f
commit c7f8aa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/proteinworkshop/embed.py b/proteinworkshop/embed.py
@@ -42,7 +42,7 @@ def embed(cfg: omegaconf.DictConfig):
     # https://pytorch.org/docs/stable/generated/torch.nn.modules.lazy.LazyModuleMixin.html#torch.nn.modules.lazy.LazyModuleMixin
     log.info("Initializing lazy layers...")
     with torch.no_grad():
-        datamodule.setup(stage="lazy_inti")  # type: ignore
+        datamodule.setup(stage="lazy_init")  # type: ignore
         batch = next(iter(datamodule.val_dataloader()))
         log.info(f"Unfeaturized batch: {batch}")
         batch = model.featurise(batch)