From 28e7e21a334001f3e62ce15875f3126af0affbd6 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Fri, 18 Oct 2024 00:37:29 +0000 Subject: [PATCH] fix --- .../librispeech_deepspeech/librispeech_jax/workload.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py index 05fdf90e7..2d46960ed 100644 --- a/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py +++ b/algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py @@ -1,5 +1,5 @@ import functools -from typing import Optional +from typing import Optional, Dict, Tuple from flax import jax_utils import jax