`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
 +from typing import List, Tuple, Optional, Dict
  import numpy as np
  import logging
  from pathlib import Path