Merge pull request #151 from relf/ft-moe

MOE improvements
SMTorg · Jun 11, 2019 · 4feae51 · 4feae51
2 parents 912e3c8 + f8b6882
commit 4feae51
Show file tree

Hide file tree

Showing 4 changed files with 167 additions and 107 deletions.
diff --git a/smt/extensions/extensions.py b/smt/extensions/extensions.py
@@ -18,11 +18,12 @@ class Extensions(object):
 
  if compiled_available:
  _surrogate_type = {
- 'KRG': KRG,'LS': LS,'QP': QP,'KPLS':KPLS,'KPLSK':KPLSK,'GEKPLS':GEKPLS,
- 'RBF':RBF,'RMTC':RMTC,'RMTB':RMTB,'IDW':IDW}
+ 'KRG': KRG, 'LS': LS, 'QP': QP, 
+ 'KPLS': KPLS, 'KPLSK': KPLSK, 'GEKPLS': GEKPLS,
+ 'RBF': RBF, 'RMTC': RMTC, 'RMTB': RMTB, 'IDW': IDW}
  else:
  _surrogate_type = {
- 'KRG': KRG,'LS': LS,'QP': QP,'KPLS':KPLS,'KPLSK':KPLSK,'GEKPLS':GEKPLS}
+ 'KRG': KRG, 'LS': LS, 'QP': QP, 'KPLS': KPLS, 'KPLSK': KPLSK, 'GEKPLS': GEKPLS}
 
  def __init__(self, **kwargs):
  """