remove unused import

mlcommons · Jan 4, 2024 · a40e085 · a40e085
1 parent 15fd5b1
commit a40e085
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/algorithmic_efficiency/workloads/ogbg/ogbg_jax/workload.py b/algorithmic_efficiency/workloads/ogbg/ogbg_jax/workload.py
@@ -3,7 +3,6 @@
 from typing import Any, Dict, Optional, Tuple
 
 from flax import jax_utils
-import flax.linen as nn
 import jax
 import jax.numpy as jnp
 import jraph