[pre-commit.ci] Add auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

mmlearn/modules/encoders/clip.py

@@ -327,7 +327,7 @@ def forward(self, inputs: dict[str, Any]) -> tuple[torch.Tensor]:
         """
         input_ids = inputs[Modalities.TEXT.name]
         attention_mask: Optional[torch.Tensor] = inputs.get(
-            "attention_mask", inputs.get(Modalities.TEXT.attention_mask, None)
+            "attention_mask", inputs.get(Modalities.TEXT.attention_mask)
         )
         position_ids = inputs.get("position_ids")
 

mmlearn/modules/encoders/text.py

@@ -161,7 +161,7 @@ def forward(self, inputs: dict[str, Any]) -> BaseModelOutput:
         outputs = self.model(
             input_ids=inputs[Modalities.TEXT.name],
             attention_mask=inputs.get(
-                "attention_mask", inputs.get(Modalities.TEXT.attention_mask, None)
+                "attention_mask", inputs.get(Modalities.TEXT.attention_mask)
             ),
             position_ids=inputs.get("position_ids"),
             output_attentions=inputs.get("output_attentions"),

projects/bioscan_clip/encoders.py

@@ -81,7 +81,7 @@ def forward(self, inputs: dict[str, Any]) -> BaseModelOutput:
         outputs = self.model(
             input_ids=inputs[Modalities.DNA.name],
             attention_mask=inputs.get(
-                "attention_mask", inputs.get(Modalities.DNA.attention_mask, None)
+                "attention_mask", inputs.get(Modalities.DNA.attention_mask)
             ),
             position_ids=inputs.get("position_ids"),
             output_attentions=inputs.get("output_attentions"),