isort

cities/modeling/model_interactions.py

@@ -3,10 +3,10 @@
 from typing import Optional
 
 import dill
-import pyro
-import pyro.distributions as dist
 import torch
 
+import pyro
+import pyro.distributions as dist
 from cities.modeling.modeling_utils import (
     prep_wide_data_for_inference,
     train_interactions_model,

cities/modeling/modeling_utils.py

@@ -2,18 +2,18 @@
 
 import matplotlib.pyplot as plt
 import pandas as pd
-import pyro
 import torch
-from pyro.infer import SVI, Trace_ELBO
-from pyro.infer.autoguide import AutoNormal
-from pyro.optim import Adam # type: ignore
 from scipy.stats import spearmanr
 
+import pyro
 from cities.utils.data_grabber import (
     DataGrabber,
     list_available_features,
     list_tensed_features,
 )
+from pyro.infer import SVI, Trace_ELBO
+from pyro.infer.autoguide import AutoNormal
+from pyro.optim import Adam  # type: ignore
 
 
 def drop_high_correlation(df, threshold=0.85):

cities/queries/causal_insight.py

@@ -5,10 +5,10 @@
 import numpy as np
 import pandas as pd
 import plotly.graph_objects as go
-import pyro
 import torch
 from sklearn.preprocessing import StandardScaler
 
+import pyro
 from cities.modeling.model_interactions import model_cities_interaction
 from cities.modeling.modeling_utils import prep_wide_data_for_inference
 from cities.utils.cleaning_utils import (