From 44eafb6c92c97451d6c3face8bd02401113eb491 Mon Sep 17 00:00:00 2001
From: Edward Yang <ezyang@meta.com>
Date: Thu, 21 Nov 2024 11:08:51 -0800
Subject: [PATCH] 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
---
 pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py b/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py
index 9bc350a..e74f842 100644
--- 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()