remove accidental LiteralString import

This commit is contained in:
julius 2023-11-07 18:52:51 +01:00
parent c6f718a1d4
commit 78f8b6cc00
Signed by untrusted user who does not match committer: julius
GPG Key ID: 8AA3791362A8084A

View File

@ -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