From 78f8b6cc00e8140bcc83439cdff8fefd694fbf57 Mon Sep 17 00:00:00 2001 From: julius Date: Tue, 7 Nov 2023 18:52:51 +0100 Subject: [PATCH] remove accidental LiteralString import --- src/prototorch/models/glvq.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/prototorch/models/glvq.py b/src/prototorch/models/glvq.py index 7709e93..cc187e7 100644 --- a/src/prototorch/models/glvq.py +++ b/src/prototorch/models/glvq.py @@ -1,7 +1,5 @@ """Models based on the GLVQ framework.""" -from typing import LiteralString - import torch from numpy.typing import NDArray from prototorch.core.competitions import wtac @@ -255,7 +253,7 @@ class GMLMLVQ(GLVQ): masks = kwargs.get("masks") for i, _mask in enumerate(masks): self.register_buffer(f"_mask_{i}", _mask) - self._masks = [self.__getattr__(f"_mask_{i}") for i,_ in enumerate(masks)] + self._masks = [self.__getattr__(f"_mask_{i}") for i, _ in enumerate(masks)] self._omegas = ParameterList([LLTI(mask).generate(1, 1) for mask in masks]) @property