From b6be7691627b2ef79d8e5a9634eb38012fe65be8 Mon Sep 17 00:00:00 2001
From: Ethan Harris <ethanwharris@gmail.com>
Date: Mon, 15 Nov 2021 12:51:44 +0000
Subject: [PATCH] Fix

---
 flash/image/classification/adapters.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flash/image/classification/adapters.py b/flash/image/classification/adapters.py
index 922e2666ab..8ed89d6fcc 100644
--- 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: