From 7997cdf170011458c348f50bf070e0ab1fe72c78 Mon Sep 17 00:00:00 2001 From: Dawith Date: Tue, 28 Oct 2025 12:46:04 -0400 Subject: [PATCH] removed an unnecessary function --- decodertest.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/decodertest.py b/decodertest.py index 70188cc..e150970 100644 --- a/decodertest.py +++ b/decodertest.py @@ -29,13 +29,6 @@ import train.decoder_train as dt from visualize.plot import spectra_plot -def build_dict(df, key): - df = df.select(key, f"{key}_str").distinct() - - return df.rdd.map( - lambda row: (str(np.argmax(row[key])), row[f"{key}_str"]) - ).collectAsMap() - def multi_gpu(): devices = jax.devices("gpu") mesh = keras.distribution.DeviceMesh(