julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove accidental LiteralString import

Browse Source
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
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/prototorch/models/glvq.py
View File

@ -1,7 +1,5 @@
"""Models based on the GLVQ framework.""" """Models based on the GLVQ framework."""
from typing import LiteralString
import torch import torch
from numpy.typing import NDArray from numpy.typing import NDArray
from prototorch.core.competitions import wtac from prototorch.core.competitions import wtac
@ -255,7 +253,7 @@ class GMLMLVQ(GLVQ):
masks = kwargs.get("masks") masks = kwargs.get("masks")
for i, _mask in enumerate(masks): for i, _mask in enumerate(masks):
self.register_buffer(f"_mask_{i}", _mask) 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]) self._omegas = ParameterList([LLTI(mask).generate(1, 1) for mask in masks])
@property @property