Fix PYRE_MISSING_ANNOTATIONS issues in fbcode/reagent/test/training/test_dqn_base.py

generatedunixname1401739305002972 · meta-codesync[bot] · commit 886a08829f8d · 2026-05-05T04:03:29.000-07:00
Differential Revision: D103829569

fbshipit-source-id: b437f77ed17136bb2c5c367b7e7af935a5f914e6
diff --git a/reagent/test/training/test_dqn_base.py b/reagent/test/training/test_dqn_base.py
@@ -159,7 +159,7 @@ def test_boost_rewards(self):
             torch.equal(boosted_reward, torch.tensor([[3.0], [2.0], [3.0]]))
         )
 
-    def test__initialize_cpe(self):
+    def test__initialize_cpe(self) -> None:
         reward_network = self.reward_network
         q_network_cpe = self.q_network_cpe
         q_network_cpe_target = self.q_network_cpe_target
@@ -184,7 +184,7 @@ def test__initialize_cpe(self):
         )
         self.assertIsNone(trainer.reward_network)
 
-    def test__initialize_cpe_extra_metrics(self):
+    def test__initialize_cpe_extra_metrics(self) -> None:
         reward_options = RewardOptions(
             metric_reward_values={"metric_a": 2, "metric_b": -2}
         )

Original file line number	Diff line number	Diff line change
`@@ -159,7 +159,7 @@ def test_boost_rewards(self):`
`159`	`159`	`torch.equal(boosted_reward, torch.tensor([[3.0], [2.0], [3.0]]))`
`160`	`160`	`)`
`161`	`161`
`162`		`- def test__initialize_cpe(self):`
	`162`	`+ def test__initialize_cpe(self) -> None:`
`163`	`163`	`reward_network = self.reward_network`
`164`	`164`	`q_network_cpe = self.q_network_cpe`
`165`	`165`	`q_network_cpe_target = self.q_network_cpe_target`
`@@ -184,7 +184,7 @@ def test__initialize_cpe(self):`
`184`	`184`	`)`
`185`	`185`	`self.assertIsNone(trainer.reward_network)`
`186`	`186`
`187`		`- def test__initialize_cpe_extra_metrics(self):`
	`187`	`+ def test__initialize_cpe_extra_metrics(self) -> None:`
`188`	`188`	`reward_options = RewardOptions(`
`189`	`189`	`metric_reward_values={"metric_a": 2, "metric_b": -2}`
`190`	`190`	`)`