`d4ec02a`

fix: Add missing Dict import to hybrid_trainer.py

The train_hybrid_models command was failing with:
NameError: name 'Dict' is not defined

Added Dict to the typing imports in hybrid_trainer.py line 16.

Authored by Claude <noreply@anthropic.com> 6 months ago

Committed by mfwolffe

SHA: d4ec02af9879880af514c130ed12c6d243bfdbc7
Parents: ed0cf04
Tree: 867268a

1 changed file

Status	File	+	-
M	`backend/jubjub/jubjubword/hybrid_trainer.py`	1	1

backend/jubjub/jubjubword/hybrid_trainer.pymodified


 import torch.nn as nn
 import torch.optim as optim
 from torch.utils.data import Dataset, DataLoader
+from typing import List, Tuple, Optional, Dict
 import numpy as np
 import logging
 from pathlib import Path

`@@ -13,7 +13,7 @@` import torch
13	import torch.nn as nn	13	import torch.nn as nn
14	import torch.optim as optim	14	import torch.optim as optim
15	from torch.utils.data import Dataset, DataLoader	15	from torch.utils.data import Dataset, DataLoader
16	-from typing import List, Tuple, Optional	16	+from typing import List, Tuple, Optional, Dict
17	import numpy as np	17	import numpy as np
18	import logging	18	import logging
19	from pathlib import Path	19	from pathlib import Path