address failing lininteg_ext.py

mfem · Feb 2, 2024 · 2d54589 · 2d54589
1 parent dde3e01
commit 2d54589
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/mfem/common/generate_lininteg_ext.py b/mfem/common/generate_lininteg_ext.py
@@ -5,11 +5,17 @@
 for line in fid.readlines():
     if line.startswith("class"):
         cname = (line.split(' ')[1]).split('(')[0]
-         
+
     if line.startswith("    def __init__"):
         pp = ""
         if line.find("*args") != -1:
             pp = "    self._coeff = args"
+        elif line.find("self, seed_=0") != -1:
+            pp = "    self._coeff = QG"
+        elif line.find("self, vqfc, ir") != -1:
+            pp = "    self._coeff = (vqfc, ir)"
+        elif line.find("self, qfc, ir") != -1:
+            pp = "    self._coeff = (qfc, ir)"
         elif line.find(", QG") != -1:
             pp = "    self._coeff = QG"
         elif line.find(", QF)") != -1: