fix

Signed-off-by: Hemil Desai <hemild@nvidia.com>

Author: hemildesai

tests/unit/models/automodel/test_automodel_train.py

@@ -18,6 +18,11 @@
 import pytest
 import torch
 
+try:
+    import nemo_automodel  # noqa: F401
+except ImportError:
+    pytest.skip("nemo_automodel not available", allow_module_level=True)
+
 from nemo_rl.distributed.batched_data_dict import BatchedDataDict
 from nemo_rl.models.automodel.data import (
     ProcessedInputs,