Skip to content

Commit afad47e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent dd4d6e6 commit afad47e

1 file changed

Lines changed: 12 additions & 15 deletions

File tree

use_cases/multiGPU.ipynb

Lines changed: 12 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -7,16 +7,13 @@
77
"metadata": {},
88
"outputs": [],
99
"source": [
10-
"import os\n",
11-
"import subprocess\n",
12-
"import pytest\n",
13-
"from mudata import MuData\n",
10+
"import tempfile\n",
11+
"\n",
1412
"import scanpy as sc\n",
1513
"import scvi\n",
16-
"from scvi.model import SCVI\n",
17-
"import tempfile\n",
1814
"import seaborn as sns\n",
19-
"import torch"
15+
"import torch\n",
16+
"from scvi.model import SCVI"
2017
]
2118
},
2219
{
@@ -270,13 +267,13 @@
270267
"source": [
271268
"print(\"multi GPU SCVI train\")\n",
272269
"model.train(\n",
273-
" max_epochs=100,\n",
274-
" check_val_every_n_epoch=1,\n",
275-
" accelerator=\"gpu\",\n",
276-
" devices=-1,\n",
277-
" datasplitter_kwargs=datasplitter_kwargs,\n",
278-
" strategy=\"ddp_notebook_find_unused_parameters_true\",\n",
279-
" )\n",
270+
" max_epochs=100,\n",
271+
" check_val_every_n_epoch=1,\n",
272+
" accelerator=\"gpu\",\n",
273+
" devices=-1,\n",
274+
" datasplitter_kwargs=datasplitter_kwargs,\n",
275+
" strategy=\"ddp_notebook_find_unused_parameters_true\",\n",
276+
")\n",
280277
"print(\"done\")"
281278
]
282279
},
@@ -490,7 +487,7 @@
490487
" color=[\"donor\", \"cell_source\"],\n",
491488
" ncols=2,\n",
492489
" frameon=False,\n",
493-
")\n"
490+
")"
494491
]
495492
},
496493
{

0 commit comments

Comments
 (0)