From 9a1839a33fb853da7d7a64ec53a282966c0b21bd Mon Sep 17 00:00:00 2001
From: Masahiro Wada <argon.argon.argon@gmail.com>
Date: Wed, 6 Apr 2022 14:58:02 +0000
Subject: [PATCH] Remove unused head argument

---
 flash/image/embedding/strategies/default.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/flash/image/embedding/strategies/default.py b/flash/image/embedding/strategies/default.py
index 357223cd828..32650f3695d 100644
--- a/flash/image/embedding/strategies/default.py
+++ b/flash/image/embedding/strategies/default.py
@@ -28,11 +28,10 @@ class DefaultAdapter(Adapter):
 
     required_extras: str = "image"
 
-    def __init__(self, backbone: torch.nn.Module, head: torch.nn.Module):
+    def __init__(self, backbone: torch.nn.Module):
         super().__init__()
 
         self.backbone = backbone
-        self.head = head
 
     @classmethod
     @catch_url_error
@@ -41,10 +40,9 @@ def from_task(
         *args,
         task: AdapterTask,
         backbone: torch.nn.Module,
-        head: torch.nn.Module,
         **kwargs,
     ) -> Adapter:
-        adapter = cls(backbone, head)
+        adapter = cls(backbone)
         adapter.__dict__["_task"] = task
         return adapter