From 95d605118fd6b0e35ff1b8b58b7baa40c6586346 Mon Sep 17 00:00:00 2001
From: carsen-stringer <carsen.stringer@gmail.com>
Date: Tue, 15 Oct 2024 22:08:38 -0400
Subject: [PATCH] Update README.md

---
 README.md | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index b354508..e4d49a3 100644
--- a/README.md
+++ b/README.md
@@ -126,16 +126,14 @@ Short example code snippet for running rastermap:
 ```
 import numpy as np
 import matplotlib.pyplot as plt
-from rastermap import Rastermap, utils
-from scipy.stats import zscore
+from rastermap import Rastermap
 
 # spks is neurons by time
 spks = np.load("spks.npy").astype("float32")
-spks = zscore(spks, axis=1)
 
 # fit rastermap
 model = Rastermap(n_PCs=200, n_clusters=100, 
-                  locality=0.75, time_lag_window=5).fit(spks, compute_X_embedding=True)
+                  locality=0.75, time_lag_window=5).fit(spks)
 y = model.embedding # neurons x 1
 isort = model.isort