update rom main driver again

LLNL · Jun 14, 2024 · 0b19d9b · 0b19d9b
1 parent 5c75b60
commit 0b19d9b
Showing 1 changed file with 7 additions and 12 deletions.
diff --git a/src/rom_main.cc b/src/rom_main.cc
@@ -50,7 +50,6 @@ int main(int argc, char** argv)
     std::string input_filename("");
     std::string lrs_filename;
     std::string constraints_filename("");
-    bool tcheck = false;
 
     float total_spin = 0.;
     bool with_spin   = false;
@@ -64,7 +63,7 @@ int main(int argc, char** argv)
     if (MPIdata::onpe0)
     {
         read_config(argc, argv, vm, input_filename, lrs_filename,
-            constraints_filename, total_spin, with_spin, tcheck);
+            constraints_filename, total_spin, with_spin);
     }
 
     MGmol_MPI::setup(comm, std::cout, with_spin);
@@ -97,17 +96,13 @@ int main(int argc, char** argv)
             mgmol = new MGmol<ExtendedGridOrbitals>(global_comm, *MPIdata::sout,
                 input_filename, lrs_filename, constraints_filename);
 
-        if (!tcheck)
-        {
-            if (ct.isLocMode())
-                readRestartFiles<LocGridOrbitals>(mgmol);
-            else
-                readRestartFiles<ExtendedGridOrbitals>(mgmol);
-        }
+        mgmol->setup();
+
+        if (ct.isLocMode())
+            readRestartFiles<LocGridOrbitals>(mgmol);
         else
-        {
-            *MPIdata::sout << " Input parameters OK\n";
-        }
+            readRestartFiles<ExtendedGridOrbitals>(mgmol);
+
         delete mgmol;
 
     } // close main scope