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
1 changed files with 1 additions and 3 deletions
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