Commit 04dd8634 authored by yuedong0607's avatar yuedong0607
Browse files

run ruff check and format

parent 2b983e18

catalog/C10_Catalog.py

+0 −7
Original line number Diff line number Diff line 
import os

import galsim

import random

import copy

import numpy as np

import h5py as h5

import healpy as hp

import astropy.constants as cons

import traceback

from astropy.coordinates import spherical_to_cartesian

from astropy.table import Table

from scipy import interpolate

from datetime import datetime



from observation_sim.mock_objects import CatalogBase, Star, Galaxy, Quasar, ExtinctionMW

from observation_sim.mock_objects._util import (

    tag_sed,

    getObservedSED,

    getABMAG,

    integrate_sed_bandpass,

    comoving_dist,

)

from observation_sim.astrometry.Astrometry_util import on_orbit_obs_position



# (TEST)

from astropy.cosmology import FlatLambdaCDM

from astropy import constants

from astropy import units as U

from astropy.coordinates import SkyCoord

from astropy.io import fits

import astropy.constants as atcons

catalog/C9_Catalog.py

+0 −5
Original line number Diff line number Diff line 
import os

import galsim

import random

import copy

import numpy as np

import h5py as h5

import healpy as hp

import astropy.constants as cons

import traceback

from astropy.coordinates import spherical_to_cartesian

from astropy.table import Table

from scipy import interpolate

from datetime import datetime



from observation_sim.mock_objects import CatalogBase, Star, Galaxy, Quasar, ExtinctionMW

from observation_sim.mock_objects._util import (

    tag_sed,

    getObservedSED,

    getABMAG,

    integrate_sed_bandpass,

    comoving_dist,

)

from observation_sim.astrometry.Astrometry_util import on_orbit_obs_position



# (TEST)

from astropy.cosmology import FlatLambdaCDM

from astropy import constants

from astropy import units as U

from astropy.coordinates import SkyCoord

from astropy.io import fits

catalog/Catalog_example.py

+1 −1
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ import astropy.constants as cons 
from astropy.table import Table

from scipy import interpolate



from observation_sim.mock_objects import CatalogBase, Star, Galaxy, Quasar

from observation_sim.mock_objects import CatalogBase, Star





class Catalog(CatalogBase):

catalog/testCat_fits.py

+2 −10
Original line number Diff line number Diff line
@@ -5,29 +5,21 @@ import numpy as np 
import h5py as h5

import healpy as hp

import astropy.constants as cons

import traceback

from astropy.coordinates import spherical_to_cartesian

from astropy.table import Table

from scipy import interpolate

from datetime import datetime



from observation_sim.mock_objects import CatalogBase, Star, Galaxy, Quasar, Stamp

from observation_sim.mock_objects import CatalogBase, Stamp

from observation_sim.mock_objects._util import (

    tag_sed,

    getObservedSED,

    getABMAG,

    integrate_sed_bandpass,

    comoving_dist,

)

from observation_sim.astrometry.Astrometry_util import on_orbit_obs_position



import astropy.io.fits as fitsio

from observation_sim.mock_objects._util import seds, sed_assign, extAv

from observation_sim.mock_objects._util import seds, sed_assign



# (TEST)

from astropy.cosmology import FlatLambdaCDM

from astropy import constants

from astropy import units as U



try:

    import importlib.resources as pkg_resources

observation_sim/_util.py

+1 −3
Original line number Diff line number Diff line 
import numpy as np

import os

from datetime import datetime

import argparse

from astropy.time import Time



from observation_sim.config import Pointing
@@ -134,7 +132,7 @@ def make_output_pointing_dir(path_dict, config, pointing_ID=0): 




def get_shear_field(config):

    if not config["shear_setting"]["shear_type"] in ["constant", "catalog"]:

    if config["shear_setting"]["shear_type"] not in ["constant", "catalog"]:

        raise ValueError("Please set a right 'shear_method' parameter.")



    if config["shear_setting"]["shear_type"] == "constant":