more lint

Signed-off-by: Bill Nell <bnell@redhat.com>

Author: bnellnm

tests/kernels/moe/utils.py

@@ -163,6 +163,7 @@ def make_quantized_test_activations(
         a_scale = torch.stack(a_scale)
 
         if not per_act_token_quant and block_shape is None:
+            assert a_scale is not None
             a_scale = a_scale.view(E, 1, 1)
 
     return a, a_q, a_scale

tests/kernels/utils.py

@@ -1115,6 +1115,8 @@ def torch_experts(
                                                      w2_scale[i], block_shape,
                                                      out.dtype)
             else:
+                assert (a_scale is not None and w1_scale is not None
+                        and w2_scale is not None)
                 f32 = torch.float32
                 scales = a_scale if a_scale.numel() == 1 else a_scale[mask]
                 tmp1 = a[mask].to(f32) * scales