flake fixes

OpenMDAO · Jan 24, 2025 · 2545a73 · 2545a73
1 parent e459560
commit 2545a73
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/mphys/network/remote_component.py b/mphys/network/remote_component.py
@@ -1,6 +1,8 @@
 import openmdao.api as om
-import json, time, os
 import numpy as np
+import json
+import time
+import os
 
 
 class RemoteComp(om.ExplicitComponent):
@@ -326,7 +328,7 @@ def _dump_json(self, remote_dict: dict, command: str):
             if not os.path.isdir(save_dir):
                 try:
                     os.mkdir(save_dir)
-                except:
+                except Exception:
                     pass  # may have been created by now, by a parallel server
             if self._doing_derivative_evaluation(command):
                 filename = f"{save_dir}/{self.name}_{dict_type}_derivative{len(self.times_gradient)}.json"
@@ -499,4 +501,4 @@ def _receive_outputs_from_server(self):
         raise NotImplementedError
 
     def _setup_server_manager(self):
-        raise NotImplementedError
+        raise NotImplementedError
diff --git a/mphys/network/server.py b/mphys/network/server.py
@@ -61,7 +61,7 @@ def _load_the_model(self):
         self.prob = om.Problem()
         self.prob.model = self.get_om_group_function_pointer()
         if self.ignore_setup_warnings:
-            with warnings.catch_warnings(record=True) as w:
+            with warnings.catch_warnings(record=True):
                 self.prob.setup(mode="rev")
         else:
             self.prob.setup(mode="rev")
@@ -75,7 +75,7 @@ def _load_the_model(self):
 
     def _run_model(self):
         if self.ignore_runtime_warnings:
-            with warnings.catch_warnings(record=True) as w:
+            with warnings.catch_warnings(record=True):
                 self.prob.run_model()
         else:
             self.prob.run_model()
@@ -86,7 +86,7 @@ def _run_model(self):
     def _compute_totals(self):
         of, wrt = self._get_derivative_inputs_outputs()
         if self.ignore_runtime_warnings:
-            with warnings.catch_warnings(record=True) as w:
+            with warnings.catch_warnings(record=True):
                 self.derivatives = self.prob.compute_totals(of=of, wrt=wrt)
         else:
             self.derivatives = self.prob.compute_totals(of=of, wrt=wrt)
@@ -407,7 +407,7 @@ def _set_additional_inputs_into_the_server_problem(
                 self.prob.get_val(key, get_remote=True)
                 != input_dict["additional_inputs"][key]["val"]
             )
-            if type(design_changed_condition) == type(True):
+            if isinstance(design_changed_condition, bool):
                 design_changed = deepcopy(design_changed_condition)
             elif design_changed_condition.any():
                 design_changed = True
@@ -435,7 +435,7 @@ def _set_additional_constants_into_the_server_problem(
                     self.prob.get_val(key)
                     != input_dict["additional_constants"][key]["val"]
                 )
-                if type(design_changed_condition) == type(True):
+                if isinstance(design_changed_condition, bool):
                     design_changed = deepcopy(design_changed_condition)
                 elif design_changed_condition.any():
                     design_changed = True