fix check

Author: dsikka

src/llmcompressor/pipelines/basic/pipeline.py

@@ -47,7 +47,7 @@ def __call__(
         with contextlib.ExitStack() as stack:
             stack.enter_context(calibration_forward_context(model))
 
-            if dataset_args.calibrate_moe_context:
+            if dataset_args is not None and dataset_args.calibrate_moe_context:
                 moe_calibration_context(model, stack)
 
             for batch in tqdm.tqdm(dataloader, desc="Calibrating"):

src/llmcompressor/pipelines/sequential/pipeline.py

@@ -31,7 +31,6 @@ def __call__(
         model: torch.nn.Module,
         dataloader: DataLoader,
         dataset_args: "DatasetArguments",
-        calibrate_moe_context: bool = False,
     ):
         """
         Run a sequential data pipeline according to the following steps: