From e2d835cf75e46b23463ea8b0e9c00c7649d680f4 Mon Sep 17 00:00:00 2001
From: Pavel <duminpn@gmail.com>
Date: Sat, 10 Aug 2024 08:34:32 +0300
Subject: [PATCH] linter fixes

---
 functions/eq_generation.py | 25 ++++++++++++-------------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/functions/eq_generation.py b/functions/eq_generation.py
index 96b38ac..484320a 100644
--- a/functions/eq_generation.py
+++ b/functions/eq_generation.py
@@ -1,7 +1,8 @@
-import pandas as pd
 import re
-from functools import reduce
 from collections import OrderedDict
+from functools import reduce
+
+import pandas as pd
 
 
 def generate_from_matrix(adj_matrix: pd.DataFrame) -> str:
@@ -58,22 +59,20 @@ def generate_from_mermaid(mermaid_code: str) -> str:
     lines = mermaid_code.split(';')[1:-1]
     verticies = set(reduce(lambda x, y: x+y, [(i.strip()[0]+i.strip()[-1]) for i in lines]))
 
-    terms = {k : v for k, v in zip(verticies, ['']*len(verticies))}
+    terms = dict(zip(verticies, [''] * len(verticies)))
     for i in lines:
-        out_state, intensity, in_state = re.sub(r'-->|--', '', i).strip().split()
-        if not terms.get(out_state):
-            terms[out_state] = f'\\dfrac{{dp_{out_state}}}{{dt}} = -{intensity}p_{{{out_state}}}(t) '
+        out_st, intensity, in_st = re.sub(r'-->|--', '', i).strip().split()
+        if not terms.get(out_st):
+            terms[out_st] = f'\\dfrac{{dp_{out_st}}}{{dt}} = -{intensity}p_{{{out_st}}}(t) '
         else:
-            terms[out_state] += f'- {intensity}p_{{{out_state}}}(t)'
-        # print(f'dp{out_state} = -{out_state}*p{out_state}')    
+            terms[out_st] += f'- {intensity}p_{{{out_st}}}(t)' 
 
-    # in_terms = []
     for i in lines:
-        out_state, intensity, in_state = re.sub(r'-->|--', '', i).strip().split()
-        if terms[in_state] == '':
-            terms[in_state] += f'\\dfrac{{dp_{in_state}}}{{dt}} = {intensity}p_{{{out_state}}}(t) '
+        out_st, intensity, in_st = re.sub(r'-->|--', '', i).strip().split()
+        if terms[in_st] == '':
+            terms[in_st] += f'\\dfrac{{dp_{in_st}}}{{dt}} = {intensity}p_{{{out_st}}}(t) '
         else:
-            terms[in_state] += f'+ {intensity}p_{{{out_state}}}(t) '
+            terms[in_st] += f'+ {intensity}p_{{{out_st}}}(t) '
     ode = OrderedDict(sorted(terms.items()))
     ltx_code = ''