Prepare for "Fix type-safety of torch.nn.Module instances": wave 2

Summary: See D52890934 Reviewed By: malfet, r-barnes Differential Revision: D66245100 fbshipit-source-id: 019058106ac7eaacf29c1c55912922ea55894d23
facebookresearch · Nov 21, 2024 · 44eafb6 · 44eafb6
1 parent 7462526
commit 44eafb6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py b/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py
@@ -100,6 +100,7 @@ def train_via_uniform_data(
         action_result = env.step(action_ind)
         agent.observe(action_result)
 
+    # pyre-fixme[16]: `PolicyLearner` has no attribute `training_rounds`.
     agent.policy_learner.training_rounds = training_epoches * T
     agent.learn()