feat: remove `query_data` invocation, add `assetnum` parameter to `predict_run`, and refine fallback imports.

main
MrWaradana 3 weeks ago
parent 6ceb295781
commit 56ec7a1bad

@ -3,28 +3,28 @@ import time
# prefer package-relative imports, but allow running this file directly as a script # prefer package-relative imports, but allow running this file directly as a script
try: try:
from src.modules.equipment.insert_actual_data import query_data, insert_lcca_maximo_corrective_data, insert_ms_equipment_data, insert_acquisition_cost_data from .insert_actual_data import query_data, insert_lcca_maximo_corrective_data, insert_ms_equipment_data, insert_acquisition_cost_data
from src.modules.equipment.Prediksi import Prediksi, main as predict_run from .Prediksi import Prediksi, main as predict_run
from src.modules.equipment.Eac import Eac, main as eac_run from .Eac import Eac, main as eac_run
except ImportError: except ImportError:
# fallback when there's no parent package (e.g., python run.py) # fallback when there's no parent package (e.g., python run.py)
from insert_actual_data import query_data, insert_lcca_maximo_corrective_data, insert_ms_equipment_data, insert_acquisition_cost_data from insert_actual_data import query_data
from Prediksi import Prediksi, main as predict_run from Prediksi import main as predict_run
from Eac import Eac, main as eac_run from Eac import main as eac_run
# Panggil fungsi # Panggil fungsi
async def main(): async def main():
start_time = time.time() start_time = time.time()
try: # try:
await query_data() # await query_data()
except Exception as e: # except Exception as e:
print(f"Error in query_data: {str(e)}") # print(f"Error in query_data: {str(e)}")
return # return
try: try:
prediction_result = await predict_run() prediction_result = await predict_run(assetnum="A22277")
if prediction_result is False: if prediction_result is False:
print("Prediction step failed or was skipped. Skipping EAC run.") print("Prediction step failed or was skipped. Skipping EAC run.")
return return

Loading…
Cancel
Save