Skip to content

Commit

Permalink
improve imports
Browse files Browse the repository at this point in the history
  • Loading branch information
MetinSa committed Aug 15, 2021
1 parent 6cbf848 commit 42764e6
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions zodipy/zodi.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@
import numpy as np

from zodipy import models
from zodipy import _coordinates as coords
from zodipy import _integration as integ
from zodipy import simulation
from zodipy import _coordinates
from zodipy import _integration

class Zodi:
"""Interface for simulating the Zodiacal emission.
Expand Down Expand Up @@ -55,11 +55,11 @@ def __init__(
observation_times = [observation_times]

observer_locations = [
coords.get_target_coordinates(observer, time)
_coordinates.get_target_coordinates(observer, time)
for time in observation_times
]
earth_locations = [
coords.get_target_coordinates('earth', time)
_coordinates.get_target_coordinates('earth', time)
for time in observation_times
]

Expand All @@ -77,9 +77,9 @@ def __init__(
)

if integration_config == 'default':
integration_config = integ.DEFAULT
integration_config = _integration.DEFAULT
elif integration_config == 'high':
integration_config = integ.HIGH
integration_config = _integration.HIGH
else:
raise ValueError(
"Available configs are: 'default' and 'high'"
Expand Down Expand Up @@ -126,6 +126,6 @@ def get_emission(

emission = self.simulation_strategy.simulate(nside, freq)

emission = coords.change_coordinate_system(emission, coord)
emission = _coordinates.change_coordinate_system(emission, coord)

return emission if return_comps else emission.sum(axis=0)

0 comments on commit 42764e6

Please sign in to comment.