From d4576d6e53e1cafe665b604f6b2f098f687546a3 Mon Sep 17 00:00:00 2001
From: Fabrice Normandin <normandf@mila.quebec>
Date: Fri, 15 Nov 2024 20:19:07 +0000
Subject: [PATCH] Fix minuscule typing error

Signed-off-by: Fabrice Normandin <normandf@mila.quebec>
---
 project/algorithms/jax_rl_example_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/project/algorithms/jax_rl_example_test.py b/project/algorithms/jax_rl_example_test.py
index 4835d4c4..094a5143 100644
--- a/project/algorithms/jax_rl_example_test.py
+++ b/project/algorithms/jax_rl_example_test.py
@@ -649,7 +649,7 @@ def log(
         self,
         name: str,
         value: Any,
-        module: JaxRLExample,
+        module: lightning.LightningModule,
         trainer: lightning.Trainer | JaxTrainer,
         **kwargs,
     ):