From 338def693d89029b4143fa883443e4d04e565ce2 Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Sun, 5 Oct 2025 11:52:54 -0700
Subject: [PATCH] oops

---
 dreamer4/dreamer4.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dreamer4/dreamer4.py b/dreamer4/dreamer4.py
index 659577c..4c0fe41 100644
--- a/dreamer4/dreamer4.py
+++ b/dreamer4/dreamer4.py
@@ -1036,7 +1036,7 @@ class DynamicsModel(Module):
 
         self.action_learned_embed = Parameter(torch.randn(dim) * 1e-2)
 
-        self.num_task = num_tasks
+        self.num_tasks = num_tasks
         self.task_embed = nn.Embedding(num_tasks, dim)
 
         # calculate "space" seq len