From 5fd40b188e138826afd9b9a1ff270b56616638c4 Mon Sep 17 00:00:00 2001 From: mshukor Date: Sun, 1 Jun 2025 09:36:23 +0200 Subject: [PATCH] avoid import pytest --- lerobot/common/policies/smolvla/smolvlm_with_expert.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/lerobot/common/policies/smolvla/smolvlm_with_expert.py b/lerobot/common/policies/smolvla/smolvlm_with_expert.py index c4ed557fd..891ff2091 100644 --- a/lerobot/common/policies/smolvla/smolvlm_with_expert.py +++ b/lerobot/common/policies/smolvla/smolvlm_with_expert.py @@ -16,7 +16,6 @@ import copy from typing import List, Optional, Union import torch -from pytest import Cache from torch import nn from transformers import ( AutoConfig, @@ -409,7 +408,7 @@ class SmolVLMWithExpertModel(nn.Module): self, attention_mask: Optional[torch.Tensor] = None, position_ids: Optional[torch.LongTensor] = None, - past_key_values: Optional[Union[List[torch.FloatTensor], Cache]] = None, + past_key_values: Optional[List[torch.FloatTensor]] = None, inputs_embeds: List[torch.FloatTensor] = None, use_cache: Optional[bool] = None, fill_kv_cache: Optional[bool] = None,