reformat with new pre-commit after PR#39

Author: aditya0by0

chebai/__init__.py

@@ -1,7 +1,8 @@
 import os
-import torch
 from typing import Any
 
+import torch
+
 # Get the absolute path of the current file's directory
 MODULE_PATH = os.path.abspath(os.path.dirname(__file__))
 

chebai/callbacks.py

@@ -1,8 +1,8 @@
 import json
 import os
+from typing import Any, Dict, List, Literal, Union
 
 import torch
-from typing import Any, Dict, List, Union, Literal
 from lightning.pytorch.callbacks import BasePredictionWriter
 
 

chebai/callbacks/model_checkpoint.py

@@ -1,4 +1,5 @@
 import os
+
 from lightning.fabric.utilities.cloud_io import _is_dir
 from lightning.fabric.utilities.types import _PATH
 from lightning.pytorch import LightningModule, Trainer

chebai/callbacks/prediction_callback.py

@@ -1,10 +1,10 @@
-from lightning.pytorch import Trainer, LightningModule
-from lightning.pytorch.callbacks import BasePredictionWriter
-import torch
-
 import os
 import pickle
-from typing import Sequence, Any, Literal
+from typing import Any, Literal, Sequence
+
+import torch
+from lightning.pytorch import LightningModule, Trainer
+from lightning.pytorch.callbacks import BasePredictionWriter
 
 
 class PredictionWriter(BasePredictionWriter):

chebai/cli.py

@@ -1,5 +1,7 @@
 from typing import Dict, Set
+
 from lightning.pytorch.cli import LightningArgumentParser, LightningCLI
+
 from chebai.trainer.CustomTrainer import CustomTrainer
 
 

chebai/loss/bce_weighted.py

@@ -1,11 +1,11 @@
+import os
 from typing import Optional
 
+import pandas as pd
 import torch
 
 from chebai.preprocessing.datasets.base import XYBaseDataModule
 from chebai.preprocessing.datasets.pubchem import LabeledUnlabeledMixed
-import pandas as pd
-import os
 
 
 class BCEWeighted(torch.nn.BCEWithLogitsLoss):

chebai/loss/mixed.py

@@ -1,5 +1,5 @@
-from torch import nn
 import torch
+from torch import nn
 
 
 class MixedDataLoss(nn.Module):

chebai/loss/semantic.py

@@ -2,11 +2,10 @@
 import math
 import os
 import pickle
+from typing import List, Literal, Union
 
 import torch
 
-from typing import Literal, Union, List
-
 from chebai.loss.bce_weighted import BCEWeighted
 from chebai.preprocessing.datasets.chebi import ChEBIOver100, _ChEBIDataExtractor
 from chebai.preprocessing.datasets.pubchem import LabeledUnlabeledMixed

chebai/models/base.py

@@ -1,11 +1,9 @@
-from typing import Optional, Dict, Union, Any
 import logging
+from typing import Any, Dict, Optional, Union
 
 import torch
-
-from torchmetrics import Metric
-
 from lightning.pytorch.core.module import LightningModule
+from torchmetrics import Metric
 
 from chebai.preprocessing.structures import XYData
 
@@ -226,7 +224,8 @@ def _execute(
             sync_dist (bool, optional): Whether to synchronize distributed training. Defaults to False.
 
         Returns:
-            Dict[str, Union[torch.Tensor, Any]]: A dictionary containing the processed data, labels, model output, predictions, and loss (if applicable).
+            Dict[str, Union[torch.Tensor, Any]]: A dictionary containing the processed data, labels, model output,
+            predictions, and loss (if applicable).
         """
         assert isinstance(batch, XYData)
         batch = batch.to(self.device)

chebai/models/electra.py

@@ -1,13 +1,10 @@
 import logging
 from math import pi
 from tempfile import TemporaryDirectory
-
-from typing import Any, Dict, Tuple, Optional
-
-from torch import nn, Tensor
+from typing import Any, Dict, Optional, Tuple
 
 import torch
-
+from torch import Tensor, nn
 from torch.nn.utils.rnn import pad_sequence
 from transformers import (
     ElectraConfig,