Fix lint error

takuseno · Oct 18, 2024 · f5b8f27 · f5b8f27
1 parent d491b0f
commit f5b8f27
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/models/test_builders.py b/tests/models/test_builders.py
@@ -1,3 +1,5 @@
+# pylint: disable=not-an-iterable
+
 from typing import Sequence
 
 import numpy as np
@@ -174,7 +176,7 @@ def test_create_discrete_q_function(
 
     # check share_encoder
     encoder = q_funcs[0].encoder
-    for q_func in list(q_funcs.modules())[1:]:
+    for q_func in q_funcs[1:]:
         if share_encoder:
             assert encoder is q_func.encoder
         else:
@@ -215,7 +217,7 @@ def test_create_continuous_q_function(
 
     # check share_encoder
     encoder = q_funcs[0].encoder
-    for q_func in list(q_funcs.modules())[1:]:
+    for q_func in q_funcs[1:]:
         if share_encoder:
             assert encoder is q_func.encoder
         else: