Skip to content

Commit

Permalink
Move streamlit import inside function
Browse files Browse the repository at this point in the history
Allows pytest to pass on github actions
  • Loading branch information
gschlafly committed Dec 11, 2023
1 parent ac61455 commit ab436cf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion VolumeRaytraceLFM/reconstructions.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
import torch
import numpy as np
from tqdm import tqdm
import streamlit as st
import matplotlib.pyplot as plt
from VolumeRaytraceLFM.abstract_classes import BackEnds
from VolumeRaytraceLFM.birefringence_implementations import (
Expand Down Expand Up @@ -519,6 +518,7 @@ def reconstruct(self, output_dir=None, use_streamlit=False):

n_epochs = self.iteration_params['n_epochs']
if use_streamlit:
import streamlit as st
st.write("Working on these ", n_epochs, "iterations...")
my_recon_img_plot = st.empty()
my_loss = st.empty()
Expand Down

0 comments on commit ab436cf

Please sign in to comment.