This repo allows for the complete reproduction, from processed data, of all the main and supplemental figures in the manuscript Non-linear Dimensionality Reduction on Extracellular Waveforms Reveals Physiological, Functional, and Laminar Diversity in Premotor Cortex.
I'm trying to run the WaveMAP_Figures_Data.ipynb on my google colab but when importing UMAP (from umap import umap_ as umap), I keep running into the following error. I updated jax on colab but still the error persists. Any idea how I can fix this?
Thank you in advance,
SAM
`/usr/local/lib/python3.7/dist-packages/jax/_src/lib/init.py in
73 version_str = jaxlib.version.version
74 version = check_jaxlib_version(
---> 75 jax_version=jax.version.version,
76 jaxlib_version=jaxlib.version.version,
77 minimum_jaxlib_version=jax.version._minimum_jaxlib_version)
AttributeError: module 'jax' has no attribute 'version'`