indent as w was weight_enum

Author: yoshitomo-matsubara

test/test_prototype_models.py

@@ -117,6 +117,7 @@ def test_schema_meta_validation(model_fn):
 
     problematic_weights = {}
     incorrect_params = []
+    bad_names = []
     for w in weights_enum:
         missing_fields = fields - set(w.meta.keys())
         if missing_fields:
@@ -134,11 +135,10 @@ def test_schema_meta_validation(model_fn):
             if w.meta.get("num_params") != weights_enum.DEFAULT.meta.get("num_params"):
                 incorrect_params.append(w)
 
-    bad_names = []
-    for k, v in weights_enum.__members__.items():
-        # Weights are constant members, thus should be all capital
-        if isinstance(v, Weights) and not k.isupper():
-            bad_names.append(k)
+        for k, v in w.__members__.items():
+            # Weights are constant members, thus should be all capital
+            if isinstance(v, Weights) and not k.isupper():
+                bad_names.append(k)
 
     assert not problematic_weights
     assert not incorrect_params