Commits

Akeem Wells authored 76a07796018
Add v1 of sparse checkout to casatestutils

casatestutils/casatestutils/__init__.py

Modified
12 12 import fnmatch
13 13 import logging
14 14 import filecmp
15 15 import unittest
16 16 import pickle
17 17 import numbers
18 18 import operator
19 19 import subprocess
20 20 import numpy
21 21
22 -
23 -_casa5 = False
24 22 _casa6 = False
25 23 _importmpi = False
26 24 __bypass_parallel_processing = 0
27 25
28 26 # https://stackoverflow.com/questions/52580105/exception-similar-to-modulenotfounderror-in-python-2-7
29 27 try:
30 28 ModuleNotFoundError
31 29 except NameError:
32 30 ModuleNotFoundError = ImportError
33 31
34 32 def import_casamods():
35 - try:
36 - # CASA 6
37 - logging.debug("Importing CASAtools")
38 - import casatools
39 - logging.debug("Importing CASAtasks")
40 - #try:
41 - # import casatasks
42 - # from casatasks import casalog
43 - #except (ImportError, ModuleNotFoundError):
44 - # pass
45 -
46 - try:
47 - from casampi.MPIEnvironment import MPIEnvironment
48 - _importmpi = True
49 - if not MPIEnvironment.is_mpi_enabled:
50 - __bypass_parallel_processing = 1
51 - except ImportError:
52 - print("MPIEnvironment not Enabled")
53 33
54 - _casa6 = True
34 + import casatools
55 35
56 - except (ImportError, ModuleNotFoundError):
57 - # CASA 5
58 - logging.debug("Import casa6 errors. Trying casa5...")
59 - from __main__ import default
60 - from taskinit import tbtool, mstool, iatool
61 - from casa_stack_manip import stack_find, find_casa
62 -
63 - try:
64 - from mpi4casa.MPIEnvironment import MPIEnvironment
65 - _importmpi = True
66 - if not MPIEnvironment.is_mpi_enabled:
67 - __bypass_parallel_processing = 1
68 - except ImportError:
69 - print("MPIEnvironment not Enabled")
36 + try:
37 + from casampi.MPIEnvironment import MPIEnvironment
38 + _importmpi = True
39 + if not MPIEnvironment.is_mpi_enabled:
40 + __bypass_parallel_processing = 1
41 + except ImportError:
42 + print("MPIEnvironment not Enabled")
70 43
71 - casa = find_casa()
72 - if casa.has_key('state') and casa['state'].has_key('init_version') and casa['state']['init_version'] > 0:
73 - casaglobals=True
74 - casac = stack_find("casac")
75 - #casalog = stack_find("casalog")
76 - _casa5 = True
44 + _casa6 = True
77 45
78 46 _casa6tools = set([
79 47 "agentflagger", "atcafiller", "atmosphere", "calanalysis", "calibrater", "coercetype", "componentlist", "config", "constants", "coordsys", "ctuser", "functional", "image",
80 48 "imagemetadata", "imagepol", "imager", "iterbotsink", "logsink", "measures", "miriadfiller", "ms", "msmetadata", "mstransformer", "platform", "quanta", "regionmanager", "sakura",
81 49 "sdm", "simulator", "singledishms", "spectralline", "synthesisdeconvolver", "synthesisimager", "synthesisimstore", "synthesisnormalizer", "synthesisutils", "table", "typecheck", "utils",
82 50 "vlafiller", "vpmanager"])
83 51
84 52
85 53 _casa6tasks = set([
86 54 "accor", "accum", "applycal", "asdmsummary", "bandpass", "blcal", "calstat", "clearcal", "clearstat", "concat", "conjugatevis", "cvel", "cvel2",

Everything looks good. We'll let you know here if there's anything you should know about.

Add shortcut