Visitar URL original
fix: Resolve mypy [valid-type] error for TorchTensor TypeAlias by Copilot · Pull Request #5984 · feast-dev/feast · GitHub
Skip to content
Draft
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion sdk/python/feast/online_response.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

from typing import TYPE_CHECKING, Any, Dict, List, TypeAlias, Union
from typing import TYPE_CHECKING, Any, Dict, List, Union

import pandas as pd
import pyarrow as pa
Expand All @@ -23,10 +23,14 @@
from feast.type_map import feast_value_type_to_python_type

if TYPE_CHECKING:
from typing import TypeAlias

import torch

TorchTensor: TypeAlias = torch.Tensor
else:
from typing import TypeAlias

TorchTensor: TypeAlias = Any

TIMESTAMP_POSTFIX: str = "__ts"
Expand Down
Loading