diff --git a/bird_cloud_gnn/__init__.py b/bird_cloud_gnn/__init__.py index 7d38df9..224bc5e 100644 --- a/bird_cloud_gnn/__init__.py +++ b/bird_cloud_gnn/__init__.py @@ -1,4 +1,5 @@ """Documentation about bird_cloud_gnn""" + import logging diff --git a/bird_cloud_gnn/cross_validation.py b/bird_cloud_gnn/cross_validation.py index 507587d..6939386 100644 --- a/bird_cloud_gnn/cross_validation.py +++ b/bird_cloud_gnn/cross_validation.py @@ -1,5 +1,6 @@ """Helper functions for cross validation. """ + import numpy as np import pandas as pd from dgl.dataloading import GraphDataLoader diff --git a/bird_cloud_gnn/early_stopper.py b/bird_cloud_gnn/early_stopper.py index a1f3ee8..009c6a5 100644 --- a/bird_cloud_gnn/early_stopper.py +++ b/bird_cloud_gnn/early_stopper.py @@ -1,5 +1,6 @@ """Module for early stopping class """ + import numpy as np diff --git a/bird_cloud_gnn/fake.py b/bird_cloud_gnn/fake.py index 77c8f20..effd592 100644 --- a/bird_cloud_gnn/fake.py +++ b/bird_cloud_gnn/fake.py @@ -126,9 +126,9 @@ def to01(value): point_cloud.loc[ np.random.randint(0, num_points, num_points // 100), "feat2" ] = None - point_cloud.loc[ - np.random.randint(0, num_points, num_points // 20), "feat3" - ] = None + point_cloud.loc[np.random.randint(0, num_points, num_points // 20), "feat3"] = ( + None + ) if filename is not None: point_cloud.to_csv(filename, index=None) diff --git a/tests/test_cross_validation.py b/tests/test_cross_validation.py index 954e664..4d41318 100644 --- a/tests/test_cross_validation.py +++ b/tests/test_cross_validation.py @@ -1,4 +1,5 @@ """Tests for cross_validation""" + from torch import nn from bird_cloud_gnn.cross_validation import kfold_evaluate from bird_cloud_gnn.cross_validation import leave_one_origin_out_evaluate diff --git a/tests/test_gnn_model.py b/tests/test_gnn_model.py index aa95440..72bda0c 100644 --- a/tests/test_gnn_model.py +++ b/tests/test_gnn_model.py @@ -1,4 +1,5 @@ """Tests for gnn_model module""" + import torch from dgl.dataloading import GraphDataLoader from torch import nn