Fix named argument

dfilan · dfilan · commit ffd535a6ef48 · 2022-12-12T17:21:51.000-08:00
diff --git a/src/reward_preprocessing/interpret.py b/src/reward_preprocessing/interpret.py
@@ -179,7 +179,7 @@ def interpret(
         # Combine rew net with GAN.
         gan = th.load(gan_path, map_location=th.device(device))
         model_to_analyse = RewardGeneratorCombo(
-            reward_net=rew_net, generator=gan.generator
+            rew_net=rew_net, generator=gan.generator
         )
 
     model_to_analyse.eval()  # Eval for visualization.

Original file line number	Diff line number	Diff line change
`@@ -179,7 +179,7 @@ def interpret(`
`179`	`179`	`# Combine rew net with GAN.`
`180`	`180`	`gan = th.load(gan_path, map_location=th.device(device))`
`181`	`181`	`model_to_analyse = RewardGeneratorCombo(`
`182`		`- reward_net=rew_net, generator=gan.generator`
	`182`	`+ rew_net=rew_net, generator=gan.generator`
`183`	`183`	`)`
`184`	`184`
`185`	`185`	`model_to_analyse.eval() # Eval for visualization.`