Merge pull request #154 from ArnauMiro/fix-imports

Change plots import

Author: ArnauMiro

Diff for: Examples/NN/SHRED/example_03_inference_shred_cylinder.py

@@ -106,14 +106,14 @@
 
 ## Plot error bars
 #Non-scaled
-fig, _ = pyLOM.utils.plotModalErrorBars(meanMRE)
+fig, _ = pyLOM.NN.plotModalErrorBars(meanMRE)
 fig.savefig('errorbars.pdf', dpi=300, bbox_inches='tight')
 #Scaled
-fig, _ = pyLOM.utils.plotModalErrorBars(Sscale*meanMRE)
+fig, _ = pyLOM.NN.plotModalErrorBars(Sscale*meanMRE)
 fig.savefig('errorbars_scaled.pdf', dpi=300, bbox_inches='tight')
 
 ## Plot POD modes reconstruction
-fig, _ = pyLOM.utils.plotTimeSeries(time, full_pod, meanout, stdout)
+fig, _ = pyLOM.NN.plotTimeSeries(time, full_pod, meanout, stdout)
 fig.savefig('output_modes.pdf', dpi=600)
 
 

Diff for: docs/source/notebook_examples/NN/SHRED/step_01_SVD_and_sensor_cylinder.ipynb

@@ -625,7 +625,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 27,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -640,7 +640,7 @@
     }
    ],
    "source": [
-    "fig, _ = pyLOM.utils.plotModalErrorBars(MRE)"
+    "fig, _ = pyLOM.NN.plotModalErrorBars(MRE)"
    ]
   },
   {
@@ -652,7 +652,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 28,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -667,7 +667,7 @@
     }
    ],
    "source": [
-    "fig, _ = pyLOM.utils.plotTimeSeries(time, full_pod, outres)"
+    "fig, _ = pyLOM.NN.plotTimeSeries(time, full_pod, outres)"
    ]
   },
   {

Diff for: pyLOM/NN/__init__.py

@@ -11,7 +11,7 @@
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'  
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
-from ..utils.plots  import plotSnapshot
+from ..utils.plots  import plotSnapshot, plotModalErrorBars, plotTimeSeries
 
 from .pipeline      import Pipeline
 from .utils         import Dataset, MinMaxScaler, select_device, betaLinearScheduler, create_results_folder, set_seed

Diff for: pyLOM/utils/__init__.py

@@ -13,7 +13,6 @@
 from .parall import worksplit, writesplit, is_rank_or_serial, pprint
 from .mpi    import MPI_COMM, MPI_RANK, MPI_SIZE, mpi_barrier, mpi_send, mpi_recv, mpi_sendrecv, mpi_scatter, mpi_gather, mpi_reduce, mpi_bcast
 from .gpu    import gpu_device, gpu_to_cpu, cpu_to_gpu, ascontiguousarray
-from .plots  import plotModalErrorBars, plotTimeSeries
 
 
 del errors, parall