🎨 Apply black and isort to files

i-machine-think · Mar 28, 2022 · 4533347 · 4533347
1 parent 9720106
commit 4533347
Show file tree

Hide file tree

Showing 8 changed files with 42 additions and 18 deletions.
diff --git a/diagnnose/activations/activation_reader.py b/diagnnose/activations/activation_reader.py
@@ -6,9 +6,14 @@
 from torch import Tensor
 
 from diagnnose.activations.activation_index import activation_index_to_iterable
-from diagnnose.typedefs.activations import (ActivationDict, ActivationKey,
-                                            ActivationName, ActivationNames,
-                                            ActivationRanges, SelectionFunc)
+from diagnnose.typedefs.activations import (
+    ActivationDict,
+    ActivationKey,
+    ActivationName,
+    ActivationNames,
+    ActivationRanges,
+    SelectionFunc,
+)
 from diagnnose.utils.pickle import load_pickle
 
 

diff --git a/diagnnose/activations/activation_writer.py b/diagnnose/activations/activation_writer.py
@@ -6,9 +6,13 @@
 
 import dill
 
-from diagnnose.typedefs.activations import (ActivationDict, ActivationFiles,
-                                            ActivationNames, ActivationRanges,
-                                            SelectionFunc)
+from diagnnose.typedefs.activations import (
+    ActivationDict,
+    ActivationFiles,
+    ActivationNames,
+    ActivationRanges,
+    SelectionFunc,
+)
 from diagnnose.utils.pickle import dump_pickle
 
 from .activation_reader import ActivationReader

diff --git a/diagnnose/attribute/shapley_tensor.py b/diagnnose/attribute/shapley_tensor.py
@@ -1,5 +1,4 @@
-from typing import (Any, Callable, Iterable, List, Optional, Sequence, Tuple,
-                    Union)
+from typing import Any, Callable, Iterable, List, Optional, Sequence, Tuple, Union
 from warnings import warn
 
 import torch

diff --git a/diagnnose/attribute/utils.py b/diagnnose/attribute/utils.py
@@ -1,8 +1,7 @@
 import itertools
 from functools import wraps
 from math import factorial
-from typing import (Any, Callable, Iterable, List, Optional, Sequence, Tuple,
-                    Union)
+from typing import Any, Callable, Iterable, List, Optional, Sequence, Tuple, Union
 
 import torch
 from torch import Tensor

diff --git a/diagnnose/extract/extractor.py b/diagnnose/extract/extractor.py
@@ -9,8 +9,12 @@
 from diagnnose.activations import ActivationReader, ActivationWriter
 from diagnnose.activations.selection_funcs import return_all
 from diagnnose.corpus import Corpus, create_iterator
-from diagnnose.typedefs.activations import (ActivationDict, ActivationNames,
-                                            ActivationRanges, SelectionFunc)
+from diagnnose.typedefs.activations import (
+    ActivationDict,
+    ActivationNames,
+    ActivationRanges,
+    SelectionFunc,
+)
 
 if TYPE_CHECKING:
     from diagnnose.models import LanguageModel

diff --git a/diagnnose/extract/simple_extract.py b/diagnnose/extract/simple_extract.py
@@ -4,8 +4,11 @@
 from diagnnose.activations import ActivationReader
 from diagnnose.activations.selection_funcs import return_all
 from diagnnose.corpus import Corpus
-from diagnnose.typedefs.activations import (ActivationNames, RemoveCallback,
-                                            SelectionFunc)
+from diagnnose.typedefs.activations import (
+    ActivationNames,
+    RemoveCallback,
+    SelectionFunc,
+)
 from diagnnose.utils.misc import suppress_print
 
 if TYPE_CHECKING:

diff --git a/diagnnose/probe/data_loader.py b/diagnnose/probe/data_loader.py
@@ -9,8 +9,12 @@
 from diagnnose.corpus.create_labels import create_labels_from_corpus
 from diagnnose.extract import simple_extract
 from diagnnose.models import LanguageModel
-from diagnnose.typedefs.activations import (ActivationIndex, ActivationName,
-                                            ActivationNames, SelectionFunc)
+from diagnnose.typedefs.activations import (
+    ActivationIndex,
+    ActivationName,
+    ActivationNames,
+    SelectionFunc,
+)
 from diagnnose.typedefs.probe import ControlTask, DataDict, DataSplit
 
 

diff --git a/diagnnose/syntax/evaluator.py b/diagnnose/syntax/evaluator.py
@@ -6,8 +6,14 @@
 from diagnnose.typedefs.syntax import AccuracyDict, ScoresDict
 
 from .task import SyntaxEvalTask
-from .tasks import (BlimpTask, LakretzTask, LinzenTask, MarvinTask,
-                    WarstadtTask, WinobiasTask)
+from .tasks import (
+    BlimpTask,
+    LakretzTask,
+    LinzenTask,
+    MarvinTask,
+    WarstadtTask,
+    WinobiasTask,
+)
 
 task_constructors: Dict[str, Type[SyntaxEvalTask]] = {
     "blimp": BlimpTask,