From 735546a4f9dc383f2a3a99b0b4a014fe638e0a09 Mon Sep 17 00:00:00 2001 From: jayeshthk Date: Thu, 19 Jun 2025 17:33:52 +0530 Subject: [PATCH] Add List import to model.py for type hinting --- train/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train/model.py b/train/model.py index 1e4d4af..15b2f1b 100644 --- a/train/model.py +++ b/train/model.py @@ -2,7 +2,7 @@ import torch import torch.nn as nn import torch.nn.functional as F import math -from typing import Optional, Tuple +from typing import Optional, Tuple, List from dataclasses import dataclass @dataclass