Skip to content

Commit

Permalink
fixed typing
Browse files Browse the repository at this point in the history
  • Loading branch information
BalzaniEdoardo committed Dec 15, 2024
1 parent 0c3fc32 commit 5998a2d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/nemos/basis/_transformer_basis.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from __future__ import annotations

import copy
from typing import TYPE_CHECKING, List
from typing import TYPE_CHECKING, Generator

import numpy as np

Expand Down Expand Up @@ -81,7 +81,7 @@ def basis(self):
def basis(self, basis):
self._basis = basis

def _unpack_inputs(self, X: FeatureMatrix) -> List:
def _unpack_inputs(self, X: FeatureMatrix) -> Generator:
"""Unpack inputs.
Unpack horizontally stacked inputs using slicing. This works gracefully with ``pynapple``,
Expand Down

0 comments on commit 5998a2d

Please sign in to comment.