Source the screens and gene pool from assaybench

README.md

@@ -42,11 +42,11 @@ in this repository.
 Then run it on a held-out screen and score the trajectory:
 
 ```python
-from assaybench import SequentialLoop, enrichment_factor
+from assaybench import SequentialLoop, enrichment_factor, gene_universe, load_screens
 from assayloop.acquisitions.greedy_from_model import GreedyFromModel
-from assayloop.tasks import AssayBenchGeneBatchTask, gene_universe, load_screens
+from assayloop.tasks import AssayBenchGeneBatchTask
 
-screens = load_screens(target_set="public")          # the paper's 20-screen test set
+screens = load_screens("assayloop-test")             # the paper's 20-screen test set
 universe = gene_universe(screens)                    # the f2 pool, 21,147 genes
 
 screen = screens[0]