Skip to content

Commit 398e24a

Browse files
committed
Fix ruff formatting issues with imports and unused variables
1 parent e70c66e commit 398e24a

2 files changed

Lines changed: 8 additions & 5 deletions

File tree

cross_condition_robustness.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import numpy as np
2+
import tensorflow as tf
23

34
from cnsd import Dataset
45
from cnsd.diagnosis.system import CNSD
@@ -23,7 +24,6 @@ def add_awgn(signals, snr_db):
2324
noisy_signals[i] = sig + noise
2425
return noisy_signals
2526

26-
import tensorflow as tf
2727

2828
def main():
2929
np.random.seed(42)

validate_run.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
"""
1919

2020
import numpy as np
21+
import tensorflow as tf
2122

2223
from cnsd import Dataset
2324
from cnsd.causal import signal_kurtosis
@@ -79,7 +80,7 @@ def read_mat(path, label, load):
7980
dir_path = os.path.join(fault_dir, ftype, size)
8081
if not os.path.exists(dir_path):
8182
continue
82-
for root, dirs, files in os.walk(dir_path):
83+
for root, _dirs, files in os.walk(dir_path):
8384
for f in files:
8485
if f.endswith('.mat'):
8586
load = int(f.split('_')[-1].split('.')[0])
@@ -123,7 +124,7 @@ def headline_accuracy_by_verdict(report, y_true):
123124
if m.any():
124125
out[v] = {'n': int(m.sum()), 'cnn_accuracy': float(correct[m].mean())}
125126
return out
126-
import tensorflow as tf
127+
127128

128129
def main():
129130
np.random.seed(42)
@@ -230,9 +231,11 @@ def main():
230231
examples = []
231232
for idx, r in enumerate(report.records):
232233
y_true = test_data.y[idx]
233-
if r["physics_verdict"] == "CONFIRMED" and y_true > 0:
234+
if r['physics_verdict'] == 'CONFIRMED' and y_true > 0:
234235
if y_true not in seen_classes:
235-
examples.append(f' [Class {y_true}] [{r["status"]}] {r["root_cause"]["statement"]}')
236+
examples.append(
237+
f' [Class {y_true}] [{r["status"]}] {r["root_cause"]["statement"]}'
238+
)
236239
seen_classes.add(y_true)
237240
if len(examples) >= 5:
238241
break

0 commit comments

Comments
 (0)