MAINT: address Ruff issues

ComPWA · Aug 7, 2024 · 0eddb73 · 0eddb73
1 parent eeba29f
commit 0eddb73
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 12 deletions.
diff --git a/2021.11.29/tensorwaves-0.3.4.ipynb b/2021.11.29/tensorwaves-0.3.4.ipynb
@@ -55,7 +55,7 @@
     "\n",
     "import ampform\n",
     "import graphviz\n",
-    "import matplotlib\n",
+    "import matplotlib as mpl\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import qrules\n",
@@ -77,7 +77,7 @@
     "tf.get_logger().setLevel(\"WARNING\")\n",
     "warnings.filterwarnings(\"ignore\")\n",
     "\n",
-    "matplotlib.rcParams.update({\"font.size\": 14})\n",
+    "mpl.rcParams.update({\"font.size\": 14})\n",
     "np_config.enable_numpy_behavior()\n",
     "\n",
     "has_tf_gpu = bool(tf.config.list_physical_devices(\"GPU\"))\n",

diff --git a/2021.11.29/tensorwaves.ipynb b/2021.11.29/tensorwaves.ipynb
@@ -55,7 +55,7 @@
     "\n",
     "import ampform\n",
     "import graphviz\n",
-    "import matplotlib\n",
+    "import matplotlib as mpl\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import qrules\n",
@@ -77,7 +77,7 @@
     "tf.get_logger().setLevel(\"WARNING\")\n",
     "warnings.filterwarnings(\"ignore\")\n",
     "\n",
-    "matplotlib.rcParams.update({\"font.size\": 14})\n",
+    "mpl.rcParams.update({\"font.size\": 14})\n",
     "np_config.enable_numpy_behavior()\n",
     "\n",
     "has_tf_gpu = bool(tf.config.list_physical_devices(\"GPU\"))\n",

diff --git a/2021.12.13/ampform.ipynb b/2021.12.13/ampform.ipynb
@@ -457,7 +457,7 @@
    "source": [
     "expr_subs = expr.subs(mass_substitutions)\n",
     "expr_subs = round_nested(expr_subs, n_decimals=2)\n",
-    "node = [s for s in sp.preorder_traversal(expr_subs) if isinstance(s, sp.Float)][0]\n",
+    "node = next(s for s in sp.preorder_traversal(expr_subs) if isinstance(s, sp.Float))\n",
     "dot = sp.dotprint(expr_subs.subs(node, Symbol(f\"{node}\")), **graph_style)\n",
     "display(expr_subs.subs(node, sp.Symbol(Rf\"\\color{{RoyalBlue}}{node}\")))\n",
     "graphviz.Source(dot)"

diff --git a/2021.12.13/tensorwaves-caching.ipynb b/2021.12.13/tensorwaves-caching.ipynb
@@ -57,7 +57,7 @@
     "\n",
     "import ampform\n",
     "import graphviz\n",
-    "import matplotlib\n",
+    "import matplotlib as mpl\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import qrules\n",
@@ -85,7 +85,7 @@
     "tf.get_logger().setLevel(\"WARNING\")\n",
     "warnings.filterwarnings(\"ignore\")\n",
     "\n",
-    "matplotlib.rcParams.update({\"font.size\": 14})\n",
+    "mpl.rcParams.update({\"font.size\": 14})\n",
     "np_config.enable_numpy_behavior()\n",
     "\n",
     "has_tf_gpu = bool(tf.config.list_physical_devices(\"GPU\"))\n",
@@ -481,7 +481,7 @@
     "    cached_expression = cached_expressions[0]\n",
     "    cse_dict = dict(cse)\n",
     "    cse_symbols = set(cse_dict)\n",
-    "    while any([bool(expr.free_symbols & cse_symbols) for expr in cse_dict.values()]):\n",
+    "    while any(bool(expr.free_symbols & cse_symbols) for expr in cse_dict.values()):\n",
     "        cse_dict = {\n",
     "            symbol: expr.xreplace(cse_dict) for symbol, expr in cse_dict.items()\n",
     "        }\n",

diff --git a/2021.12.13/tensorwaves-optimized-expression.ipynb b/2021.12.13/tensorwaves-optimized-expression.ipynb
@@ -57,7 +57,7 @@
     "\n",
     "import ampform\n",
     "import graphviz\n",
-    "import matplotlib\n",
+    "import matplotlib as mpl\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import qrules\n",
@@ -84,7 +84,7 @@
     "tf.get_logger().setLevel(\"WARNING\")\n",
     "warnings.filterwarnings(\"ignore\")\n",
     "\n",
-    "matplotlib.rcParams.update({\"font.size\": 14})\n",
+    "mpl.rcParams.update({\"font.size\": 14})\n",
     "np_config.enable_numpy_behavior()\n",
     "\n",
     "has_tf_gpu = bool(tf.config.list_physical_devices(\"GPU\"))\n",

diff --git a/2021.12.13/tensorwaves.ipynb b/2021.12.13/tensorwaves.ipynb
@@ -55,7 +55,7 @@
     "\n",
     "import ampform\n",
     "import graphviz\n",
-    "import matplotlib\n",
+    "import matplotlib as mpl\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import qrules\n",
@@ -82,7 +82,7 @@
     "tf.get_logger().setLevel(\"WARNING\")\n",
     "warnings.filterwarnings(\"ignore\")\n",
     "\n",
-    "matplotlib.rcParams.update({\"font.size\": 14})\n",
+    "mpl.rcParams.update({\"font.size\": 14})\n",
     "np_config.enable_numpy_behavior()\n",
     "\n",
     "has_tf_gpu = bool(tf.config.list_physical_devices(\"GPU\"))\n",