Fix

Lightning-Universe · Nov 15, 2021 · b6be769 · b6be769
1 parent b898714
commit b6be769
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/image/classification/adapters.py b/flash/image/classification/adapters.py
@@ -289,7 +289,7 @@ def from_task(
                 "This is equivalent to the number of sample per label to select within a task."
             )
         adapter = cls(backbone, head, algorithm, **kwargs)
-        adapter["_task"] = task
+        adapter.__dict__["_task"] = task
         return adapter
 
     def training_step(self, batch, batch_idx) -> Any: