Skip to content

Commit b56749d

Browse files
committed
Lint fix pt 2
1 parent af93e26 commit b56749d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/tiledbsoma_ml/scvi.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
from __future__ import annotations
22

3+
import os
34
from typing import Any, Sequence
45

56
import pandas as pd
67
import torch
7-
import os
88
from lightning import LightningDataModule
99
from sklearn.preprocessing import LabelEncoder
1010
from tiledbsoma import ExperimentAxisQuery
@@ -16,9 +16,10 @@
1616
DEFAULT_DATALOADER_KWARGS: dict[str, Any] = {
1717
"pin_memory": torch.cuda.is_available(),
1818
"persistent_workers": True,
19-
"num_workers": max(((os.cpu_count() or 1) // 2), 1)
19+
"num_workers": max(((os.cpu_count() or 1) // 2), 1),
2020
}
2121

22+
2223
class SCVIDataModule(LightningDataModule): # type: ignore[misc]
2324
"""PyTorch Lightning DataModule for training scVI models from SOMA data.
2425

0 commit comments

Comments
 (0)