Skip to content

Commit

Permalink
removed an unnecessary function
Browse files Browse the repository at this point in the history
  • Loading branch information
Dawith committed Oct 28, 2025
1 parent 97ebb66 commit 7997cdf
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions decodertest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 7997cdf

Please sign in to comment.