style: lint

borisdayma · borisdayma · commit d5d442a456a0 · 2022-02-22T23:37:42.000Z
diff --git a/tools/inference/inference_pipeline.ipynb b/tools/inference/inference_pipeline.ipynb
@@ -240,7 +240,7 @@
     "import random\n",
     "\n",
     "# create a random key\n",
-    "seed = random.randint(0, 2 ** 32 - 1)\n",
+    "seed = random.randint(0, 2**32 - 1)\n",
     "key = jax.random.PRNGKey(seed)"
    ]
   },
diff --git a/tools/train/train.py b/tools/train/train.py
@@ -58,7 +58,7 @@
 )
 
 cc.initialize_cache(
-    "/home/boris/dalle-mini/jax_cache", max_cache_size_bytes=5 * 2 ** 30
+    "/home/boris/dalle-mini/jax_cache", max_cache_size_bytes=5 * 2**30
 )
 
 logger = logging.getLogger(__name__)

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@`
`58`	`58`	`)`
`59`	`59`
`60`	`60`	`cc.initialize_cache(`
`61`		`- "/home/boris/dalle-mini/jax_cache", max_cache_size_bytes=5 * 2 ** 30`
	`61`	`+ "/home/boris/dalle-mini/jax_cache", max_cache_size_bytes=5 * 2**30`
`62`	`62`	`)`
`63`	`63`
`64`	`64`	`logger = logging.getLogger(__name__)`