isort fixes

nv-morpheus · Jan 18, 2024 · 37d7ef8 · 37d7ef8
1 parent e4f9646
commit 37d7ef8
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 25 deletions.
diff --git a/examples/digital_fingerprinting/production/morpheus/dfp/modules/dfp_split_users.py b/examples/digital_fingerprinting/production/morpheus/dfp/modules/dfp_split_users.py
@@ -116,9 +116,10 @@ def generate_split_dataframes(users_df: pd.DataFrame):
         if (include_individual):
             # pylint: disable=unnecessary-comprehension
             # List comprehension is necessary here to convert to a dictionary
-            split_dataframes.update(
-                {username: user_df
-                 for username, user_df in users_df.groupby(userid_column_name, sort=False)})
+            split_dataframes.update({
+                username: user_df
+                for username, user_df in users_df.groupby(userid_column_name, sort=False)
+            })
 
         return split_dataframes
 

diff --git a/models/training-tuning-scripts/common/sequence_classifier.py b/models/training-tuning-scripts/common/sequence_classifier.py
@@ -15,16 +15,16 @@
 
 import logging
 
-import cudf
-from cudf.core.subword_tokenizer import SubwordTokenizer
 import cupy
-
 import torch
 from torch.optim import AdamW
 from torch.utils.dlpack import to_dlpack
 from tqdm import trange
 from transformers import AutoModelForSequenceClassification
 
+import cudf
+from cudf.core.subword_tokenizer import SubwordTokenizer
+
 from .dataloader import DataLoader
 from .dataset import Dataset
 

diff --git a/models/validation-inference-scripts/phishing-models/phish_bert_inference_script.py b/models/validation-inference-scripts/phishing-models/phish_bert_inference_script.py
@@ -21,6 +21,14 @@
     --output phishing-email-validation-output.jsonlines
 """
 
+import argparse
+import json
+
+import numpy as np
+import onnxruntime
+import scipy
+import torch
+
 ###########################################################################################
 # cudf imports moved before torch import to avoid the following error:
 # ImportError: /usr/lib/x86_64-linux-gnu/libstdc++.so.6: version `GLIBCXX_3.4.29' not found
@@ -29,14 +37,6 @@
 
 ###########################################################################################
 
-import argparse
-import json
-
-import numpy as np
-import onnxruntime
-import torch
-import scipy
-
 
 def infer(validationdata, vocab, model, output):
 

diff --git a/models/validation-inference-scripts/root-cause-models/root_cause_inference.py b/models/validation-inference-scripts/root-cause-models/root_cause_inference.py
@@ -22,6 +22,14 @@
     --output root-cause-validation-output.jsonlines
 """
 
+import argparse
+import json
+
+import numpy as np
+import onnxruntime
+import scipy
+import torch
+
 ###########################################################################################
 # cudf imports moved before torch import to avoid the following error:
 # ImportError: /usr/lib/x86_64-linux-gnu/libstdc++.so.6: version `GLIBCXX_3.4.29' not found
@@ -30,14 +38,6 @@
 
 ###########################################################################################
 
-import argparse
-import json
-
-import numpy as np
-import onnxruntime
-import torch
-import scipy
-
 
 def infer(
     validationdata,

diff --git a/morpheus/__init__.py b/morpheus/__init__.py
@@ -12,14 +12,15 @@
 # limitations under the License.
 """Root module for the Morpheus library."""
 
+import logging
+import os
+
 # ########################### CVE-2023-47248 Mitigation ############################
 # Import pyarrow_hotfix as early as possible to ensure that the pyarrow hotfix is applied before any code can use it
 # Can be removed after upgrading to pyarrow 14.0.1 or later (which is dictated by cudf)
 import pyarrow_hotfix
-# ##################################################################################
 
-import logging
-import os
+# ##################################################################################
 
 # Create a default null logger to prevent log messages from being propagated to users of this library unless otherwise
 # configured. Use the `utils.logging` module to configure Morpheus logging