Fix formatting

.pre-commit-config.yaml

@@ -5,10 +5,12 @@ repos:
     rev: 23.7.0
     hooks:
       - id: black
+        exclude: ^(scripts/|experiment/)
   - repo: https://github.com/pycqa/isort
     rev: 5.12.0
     hooks:
       - id: isort
+        exclude: ^(scripts/|experiment/)
         args: ["--profile", "black"]
   # - repo: https://github.com/pycqa/flake8
   #   rev: 5.0.4

neuralprocesses/data/antarctica.py

@@ -3,11 +3,12 @@
 import netCDF4 as nc
 import numpy as np
 import pandas as pd
-import torch
 from lab import B
 from plum import convert
 from wbml.data.eeg import load_full as load_eeg
 
+import torch
+
 from ..aggregate import Aggregate, AggregateInput
 from ..dist import AbstractDistribution, UniformContinuous, UniformDiscrete
 from .data import DataGenerator, apply_task

neuralprocesses/tensorflow/nn.py

@@ -3,10 +3,10 @@
 
 import lab.tensorflow as B
 import numpy as np
-import tensorflow as tf
 from plum import convert
 
 import neuralprocesses as nps
+import tensorflow as tf
 
 from .. import _dispatch
 from ..util import compress_batch_dimensions, is_framework_module

neuralprocesses/torch/nn.py

@@ -3,11 +3,11 @@
 
 import lab.torch as B
 import numpy as np
-import torch
 from plum import convert
 from wbml.util import inv_perm
 
 import neuralprocesses as nps
+import torch
 
 from .. import _dispatch
 from ..util import is_framework_module

tests/gnp/gnp.py

@@ -1,8 +1,8 @@
 import lab.torch as B
 import pytest
-import torch
 
 import neuralprocesses.gnp as gnp
+import torch
 
 # noinspection PyUnresolvedReferences
 from .util import context_set, target_set

tests/gnp/util.py

@@ -1,6 +1,7 @@
 import lab.torch as B
 import numpy.testing
 import pytest
+
 import torch
 
 __all__ = ["approx", "context_set", "target_set"]

tests/util.py

@@ -3,12 +3,12 @@
 
 import lab as B
 import pytest
-import tensorflow as tf
-import torch
 from numpy.testing import assert_allclose
 from plum import Dispatcher
 
 import neuralprocesses
+import tensorflow as tf
+import torch
 
 __all__ = ["approx", "nps", "generate_data", "remote_xfail", "remote_skip"]
 

train.py

@@ -5,15 +5,15 @@
 import warnings
 from functools import partial
 
+import experiment as exp
 import lab as B
 import numpy as np
-import torch
 import wbml.out as out
 from matrix.util import ToDenseWarning
 from wbml.experiment import WorkingDirectory
 
-import experiment as exp
 import neuralprocesses.torch as nps
+import torch
 
 __all__ = ["main"]