fixing CI

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Author: HDCharles

test/quantization/quantize_/workflows/float8/test_float8_tensor.py

@@ -512,9 +512,9 @@ def test_moe_weight_reshape_ops(self):
         dim = 128
         hidden_dim = 256
 
-        moe1 = Experts(num_experts, dim, hidden_dim, dtype, device)
-        moe2 = Experts(num_experts, dim, hidden_dim, dtype, device)
-        moe_combined = Experts(num_experts, dim, 2 * hidden_dim, dtype, device)
+        moe1 = nn.Sequential(Experts(num_experts, dim, hidden_dim, dtype, device))
+        moe2 = nn.Sequential(Experts(num_experts, dim, hidden_dim, dtype, device))
+        moe_combined = nn.Sequential(Experts(num_experts, dim, 2 * hidden_dim, dtype, device))
         input = torch.randn(batch_size, input_dim, dim, dtype=dtype, device=device)
 
         moes = [moe1, moe2]