liqian 2 years ago
parent
commit
afc9d2c2ec
2 changed files with 2 additions and 6 deletions
  1. 2 2
      config.py
  2. 0 4
      utils.py

+ 2 - 2
config.py

@@ -1108,8 +1108,8 @@ class ProductionConfig(BaseConfig):
 
 def set_config():
     # 获取环境变量 ROV_OFFLINE_ENV
-    # env = os.environ.get('ROV_OFFLINE_ENV')
-    env = 'dev'
+    env = os.environ.get('ROV_OFFLINE_ENV')
+    # env = 'dev'
     if env is None:
         # log_.error('ENV ERROR: is None!')
         return

+ 0 - 4
utils.py

@@ -477,15 +477,11 @@ def get_feature_data(project, table, features, dt):
     """获取特征数据"""
     records = get_data_from_odps(date=dt, project=project, table=table)
     feature_data = []
-    i = 0
     for record in records:
-        if i > 1000:
-            break
         item = {}
         for feature_name in features:
             item[feature_name] = record[feature_name]
         feature_data.append(item)
-        i += 1
     feature_df = pd.DataFrame(feature_data)
     return feature_df