ins-pricing 0.4.5__py3-none-any.whl → 0.5.1__py3-none-any.whl

ins_pricing/utils/profiling.py

@@ -18,6 +18,10 @@ import time
 from contextlib import contextmanager
 from typing import Optional
 
+from ins_pricing.utils import get_logger
+
+_logger = get_logger("ins_pricing.utils.profiling")
+
 try:
     import psutil
     HAS_PSUTIL = True
@@ -96,7 +100,7 @@ def profile_section(
         if logger:
             logger.log(log_level, msg)
         else:
-            print(msg)
+            _logger.log(log_level, msg)
 
 
 def get_memory_info() -> dict:
@@ -250,7 +254,7 @@ def cleanup_memory(logger: Optional[logging.Logger] = None) -> None:
     if logger:
         logger.info(msg)
     else:
-        print(msg)
+        _logger.info(msg)
 
 
 class MemoryMonitor:
@@ -301,7 +305,7 @@ class MemoryMonitor:
         if self.logger:
             self.logger.info(msg)
         else:
-            print(msg)
+            _logger.info(msg)
 
         return self
 
@@ -323,7 +327,7 @@ class MemoryMonitor:
         if self.logger:
             self.logger.info(msg)
         else:
-            print(msg)
+            _logger.info(msg)
 
         # Check threshold and cleanup if needed
         if self.threshold_gb is not None:

{ins_pricing-0.4.5.dist-info → ins_pricing-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ins_pricing
-Version: 0.4.5
+Version: 0.5.1
 Summary: Reusable modelling, pricing, governance, and reporting utilities.
 Author: meishi125478
 License: Proprietary

ins_pricing-0.5.1.dist-info/RECORD

@@ -0,0 +1,132 @@
+ins_pricing/README.md,sha256=pufz_ChEpXXfLk96prv8ciqwgyVvHE5aEZ8iAH-u6k0,3679
+ins_pricing/__init__.py,sha256=pATYokjg75DXHd2zyoct62efui98SCBO2H8B06Q68r0,4317
+ins_pricing/exceptions.py,sha256=5fZavPV4zNJ7wPC75L215KkHXX9pRrfDAYZOdSKJMGo,4778
+ins_pricing/setup.py,sha256=cbVFHNm5swws1eyPy6zTkZYH2GGwRehcJiKLAZ_2mNE,1702
+ins_pricing/cli/BayesOpt_entry.py,sha256=AMdHJim2fG1_Z6DRZopt3wMWeYwTOnmcL2_8iO490kY,2058
+ins_pricing/cli/BayesOpt_incremental.py,sha256=SerZyK2a6Lr3EBvuuay_Yq1dSQ8u2KvtKuePoLS7Y9w,36422
+ins_pricing/cli/Explain_Run.py,sha256=zDtHZ6tbLXN7-mw7yCAK4R1D4oOfHr-oYwcHhuTgjXg,1281
+ins_pricing/cli/Explain_entry.py,sha256=yWhJDIUJdbVefAa0QFBDMhwhCx42HIN4cC8ZfLEos9M,23880
+ins_pricing/cli/Pricing_Run.py,sha256=mIND374OqQMOIcEMD_XdCOtMuD_RBXuG6pQhrMyr9lE,1328
+ins_pricing/cli/__init__.py,sha256=F296f1J_tBPv33lDJQ6LaN_CPwMJTMtOuTsMof0dr2o,50
+ins_pricing/cli/bayesopt_entry_runner.py,sha256=TlA0aBOenB1L4O_nq9xo3qeUgy68CpTsjVwSdlybjlA,57023
+ins_pricing/cli/watchdog_run.py,sha256=45p-tV9pvuRczPdo8Gb-awk1ih7vnMI4XDXO56kc1Bk,6995
+ins_pricing/cli/utils/__init__.py,sha256=u3kt1B27OiuOEgw6PQN-fNs9vNiAjdPyybsRQsZkM_I,54
+ins_pricing/cli/utils/bootstrap.py,sha256=tlTgFQcnnJQftzreixHm4Ls487Zpo5zLAq1BS7BWysg,689
+ins_pricing/cli/utils/cli_common.py,sha256=Tjr20GWK3XT-KidrFCHwodrRukpaInNzsMzEgM9MgYQ,8530
+ins_pricing/cli/utils/cli_config.py,sha256=RFFulQVrZDIGSGL4PoIboWikpwJIPQxQ2iZt34Yqkfg,14752
+ins_pricing/cli/utils/evaluation_context.py,sha256=0zuDOcVzkWiuj4HyAT0psaAfEbSqpDXEGXSjxztHY6E,9485
+ins_pricing/cli/utils/import_resolver.py,sha256=xowiDSBce1VfvBpcjQSCY87c--h-O_u92dgDlSdxA7w,13697
+ins_pricing/cli/utils/notebook_utils.py,sha256=N6-SFvBc81Mr43wWKd1N7RspjHMusfiODvm9In7II9s,12673
+ins_pricing/cli/utils/run_logging.py,sha256=V3Wh2EV6c1Mo0QTvfe4hl2J4LOR6bdQsT210o__YBWk,3677
+ins_pricing/frontend/README.md,sha256=MKQuVtwpvrOKBJhcz9wgVtMc-jKERYATRNDIj-hA78Y,17551
+ins_pricing/frontend/__init__.py,sha256=mbvo89webbtirzVCNmp26FrtYmLFtuwxkmZ5tcdab2U,405
+ins_pricing/frontend/app.py,sha256=Wd-ZDXQwl3c8EhxC3qWq3-Oe91PhcbEiWXT6ofrEROY,42006
+ins_pricing/frontend/config_builder.py,sha256=fXgzNpzriEymB4oRl0MY-O5_zSe5fQTTQj5q-61stSw,14181
+ins_pricing/frontend/example_config.json,sha256=2YcBo9QtC63EKbnCsrBKtqo08--K2WFPqjN-pfXXKnA,1102
+ins_pricing/frontend/example_workflows.py,sha256=-yyoEucItAdVMVnmt6X7oDldaAqe5bWdrstMoNpZOxo,36815
+ins_pricing/frontend/ft_workflow.py,sha256=CWrviErHEjZ2NPibIbPh69AYKPVkn-mXY7uncPihn-0,11206
+ins_pricing/frontend/runner.py,sha256=U28q2n4ky1B2TF0wdkolbJxRse15aNZbq7liQN9P9kU,11428
+ins_pricing/governance/README.md,sha256=XnXLS5RPzWhEiicJ3WtGmpN935jppHhPftA9Lo2DPnQ,511
+ins_pricing/governance/__init__.py,sha256=kCVZ_eNJnVNCFtefJBFz78KZxaR41IU1L2JoefiojiU,625
+ins_pricing/governance/approval.py,sha256=cjJQjU1ziR-d-9wVSXyMyX6S5zijJqDWERZNxjqGAUE,2879
+ins_pricing/governance/audit.py,sha256=f0aw-LaOxH5NGzxwczeLrGMJcxO-JDRn99BpI55KRn4,1040
+ins_pricing/governance/registry.py,sha256=2uxQL6qMGY5IYWJti9MpaV_auvL--piJaasFrX20ghk,3139
+ins_pricing/governance/release.py,sha256=9s-6V41RFaJ-z0V0RFw3_xgrk77BYfun0WN_w3uspqc,4946
+ins_pricing/modelling/README.md,sha256=eSi6OoVjz-609_XFZ2yXWfOGuK00mb9x5EYL3PgXDEs,2151
+ins_pricing/modelling/__init__.py,sha256=1Z_fDHMpuGYOGh4s662mHh4EGMQkYG5hH8Fcg5Aln3I,4366
+ins_pricing/modelling/evaluation.py,sha256=tgmQ-7RHhOkzPCLHtfm4HHNLJVwxYaFExnCVjoBqoeM,4154
+ins_pricing/modelling/bayesopt/README.md,sha256=9WBmvcEupOYUSA8LNIUHxs9rNCoEfT0Bq-gewpsAQZw,2544
+ins_pricing/modelling/bayesopt/__init__.py,sha256=wkTYjfU9Oi6sNCrn4pmF1QevEc2f_EF6k_PIiCYt-aY,1555
+ins_pricing/modelling/bayesopt/config_components.py,sha256=Pr3F3LNJTBwHoLegsYyrC8OrkQsbuSL4yQDEySQf5dU,12823
+ins_pricing/modelling/bayesopt/config_preprocess.py,sha256=xwC7SNqs5suMnhuB-qZEyfTcJHLm2OUFUShOvzYjpgk,24629
+ins_pricing/modelling/bayesopt/core.py,sha256=318JwzECkgjW6jpMQuT0bDZY7f-Gn4u3wBraJ9BlFCk,47316
+ins_pricing/modelling/bayesopt/model_explain_mixin.py,sha256=8LX72wJrzkyINI0AOE5labGhsH2VT1muIRDTiHQ2JQ4,11973
+ins_pricing/modelling/bayesopt/model_plotting_mixin.py,sha256=3y-fxn9eP9RJNBLRRWLi1QJFKerww7KtEz8Z-LUh0Uw,19413
+ins_pricing/modelling/bayesopt/models/__init__.py,sha256=gFMAdElw08bnSM2qldv3IT8O6NSlWEUevHUfFrVtYGM,875
+ins_pricing/modelling/bayesopt/models/model_ft_components.py,sha256=Ij9aX3vrKZJGbRTSBEz9UNyPCQ-QV4Nxy1HbpbFnzDo,12443
+ins_pricing/modelling/bayesopt/models/model_ft_trainer.py,sha256=hvJxMma9Ug6ssvi9tMUD0coEYdPj4fuprRGdQDA59KI,41042
+ins_pricing/modelling/bayesopt/models/model_gnn.py,sha256=CfgIwD1DQ1V6_1p0HAaPzjT9M5XwvHU9XGptu4sCw04,34041
+ins_pricing/modelling/bayesopt/models/model_resn.py,sha256=mstCB3ERb-FGGP26KAd9FTQ9K8vHoKYTVcQIafW8AWg,18361
+ins_pricing/modelling/bayesopt/trainers/__init__.py,sha256=O7u5452eQV9da6jBJRhjn_KkFLe4APGTq1x018wMqdM,734
+ins_pricing/modelling/bayesopt/trainers/trainer_base.py,sha256=0oT9m_cZ12FjbInIHyWXKnbT0tREQoNKwPtkuWg0aSI,57593
+ins_pricing/modelling/bayesopt/trainers/trainer_ft.py,sha256=jEQy6qxBqDT-Y-YloYHmWFUwRy0mXUuH9p4OHRpFVpg,37368
+ins_pricing/modelling/bayesopt/trainers/trainer_glm.py,sha256=LPciwyuvtEvj0h0RGRNxGyL9ArH15ByiXRwh8V-TdpQ,8320
+ins_pricing/modelling/bayesopt/trainers/trainer_gnn.py,sha256=EwGPD1kRtQSE6FxZdsQv-UStMl8tw7Ts5YIk6yONzlY,14973
+ins_pricing/modelling/bayesopt/trainers/trainer_resn.py,sha256=0cSgdM-mcxTjq0r3kqIXyDjKMuoIwg5U3hMcXWmz0TA,12627
+ins_pricing/modelling/bayesopt/trainers/trainer_xgb.py,sha256=GWW9kPD4hNEIASFNTB8rPh7gDp4w6yrIuNVMDee1SkA,21474
+ins_pricing/modelling/bayesopt/utils/__init__.py,sha256=SnFPs_g4eut2h2RrBjOO40Vh-tBBTCWOIHPdm7vMfqk,1553
+ins_pricing/modelling/bayesopt/utils/constants.py,sha256=SNC5NJ1FfRQ-iuNmM8xHlaoaFgNpy1vnS_QOqNbJYRE,485
+ins_pricing/modelling/bayesopt/utils/distributed_utils.py,sha256=jSiDyC7p9SPwllDkCZoLIJeijpLvGcXvQ9l9l2jA6K4,6180
+ins_pricing/modelling/bayesopt/utils/io_utils.py,sha256=2Bg0fz1dngmBU8G8GZp3-CGLIGyhAxeMxwzIXYRINMc,224
+ins_pricing/modelling/bayesopt/utils/losses.py,sha256=m4moKKwfKFl6yDIa9SGQV1u1sgX5MpKoSDF2KI85wyc,634
+ins_pricing/modelling/bayesopt/utils/metrics_and_devices.py,sha256=-TKUHBEumWC4UssZIuthr5yzRBiKFivYXvI9-rCIvxg,317
+ins_pricing/modelling/bayesopt/utils/torch_trainer_mixin.py,sha256=-iVSq6JCGNJaSsyKw-RJCmQ3kufTz_Q3OHINfABYBG8,26737
+ins_pricing/modelling/explain/__init__.py,sha256=WU4Fy0mEvqnmpX-aalKktj1goo9tnKCht9WOrWmZAtI,1342
+ins_pricing/modelling/explain/gradients.py,sha256=9TqCws_p49nFxVMcjVxe4KCZ7frezeL0uV_LCdoM5yo,11088
+ins_pricing/modelling/explain/metrics.py,sha256=9Dh-RcmV1uUlo8eacswsGeNTxTBRwSnjFHncFUFZZ8g,473
+ins_pricing/modelling/explain/permutation.py,sha256=-_6KRkMA1EKHXrfCVs6ohjS9pI65af5zfWKSLLYHZkc,9599
+ins_pricing/modelling/explain/shap_utils.py,sha256=70zRIHPPdoECFOFQeBTRxLZF-6sjaGJBNMIRS4_kmVI,10462
+ins_pricing/modelling/plotting/__init__.py,sha256=AfqRYacai9UzMPBy4xQk3PdVTbCwT31saInSOJKIbeM,1398
+ins_pricing/modelling/plotting/common.py,sha256=_kFq7JMA0LnKIp4bqAFvr-24VaHjj9pegDMm1qP9_7Y,1439
+ins_pricing/modelling/plotting/compat.py,sha256=rmOct4rrjIX7EFodMfWnDtd8D4qE1inNTMvE9FUWa9k,8039
+ins_pricing/modelling/plotting/curves.py,sha256=AyNeESaXOIN6efd3K0vAQ7hmDU4ifJNq6lx_9e3Xi7c,19000
+ins_pricing/modelling/plotting/diagnostics.py,sha256=iHvZY1zCnFRbSczVluMYYRkkQXc-mwo0Z1DXyL3XMkI,5357
+ins_pricing/modelling/plotting/geo.py,sha256=jQogaHt57W1-f8vzeGUdWE1oON2k4WVZKf2qoOM2Ct8,11401
+ins_pricing/modelling/plotting/importance.py,sha256=j5FkPJH9FA6mJHMUfNkqU4Yi7IcqdGdIpcG-d76ORiw,3817
+ins_pricing/pricing/README.md,sha256=PEcyw5oDkqJHOqnNdzBdbbpZwG4SOlnhMwY-owwQ0GI,1064
+ins_pricing/pricing/__init__.py,sha256=hPgQP4IyhLfP1hEC5XckNzYmByhYEdlqw1D83X0hC8s,1022
+ins_pricing/pricing/calibration.py,sha256=cx9fbDoOnNEMGPH6Js1EDMfVwy9J7zf_90yuNwD7W6I,6196
+ins_pricing/pricing/data_quality.py,sha256=8FecBE60cABsTZE7HETuoKCEOXIrirGAFgg5wQCZrmU,4043
+ins_pricing/pricing/exposure.py,sha256=rw8kKZ_1QdeGTCm13ck9NXrRBPt6TunxRw2s_qkHYkg,2575
+ins_pricing/pricing/factors.py,sha256=d4wTWwqZHAKtL9ZNoicEmfAQSxQ7a-XKhXBpoynfH2o,6576
+ins_pricing/pricing/monitoring.py,sha256=GnfF2g1C9KzDks9ELBykfAd0zzVBUvjuTWoffa7aGbI,613
+ins_pricing/pricing/rate_table.py,sha256=llDW95i7gR6cCtGFwcGqgpgFvOOPCURaJWmuQw1oce8,2473
+ins_pricing/production/__init__.py,sha256=4PkwZFR3qmTXyY7g3xELqobC1KcFcozJaJ95hPPB9to,1231
+ins_pricing/production/drift.py,sha256=q_oE_h2NbVETTBkh9QUu8Y68ERuFFcrfKpOb3zBcvsA,383
+ins_pricing/production/inference.py,sha256=RaTQAI92XGuvb0cU224EEfQTzGs8ODshwCSFfkAyKog,27286
+ins_pricing/production/monitoring.py,sha256=0quH2bvsDIHefVxwEE4lRIHC61t51fVEo1x6bdfhQdw,4144
+ins_pricing/production/preprocess.py,sha256=cl20X0rVcKNCjVJswB8SdHffMgox6Qga4Ac29L6pW5g,9404
+ins_pricing/production/scoring.py,sha256=yFmMmbYb7w_RC4uZOCMnAjLMRcjXQWIuT1nsfu-bwuc,1379
+ins_pricing/reporting/README.md,sha256=kTVdB6pNewwh1HlCHrI2SzWTgprtQoQprLRQ2qLdgNA,486
+ins_pricing/reporting/__init__.py,sha256=9MIGXjLrmW4lRunRz3dGhdeEGBkYJbrOO07C1ZPl6Ck,295
+ins_pricing/reporting/report_builder.py,sha256=53ZFqGUx2isAoigT5IDwvXkek67zN7-6IgKeGpJhO7c,2241
+ins_pricing/reporting/scheduler.py,sha256=9koG_1cmWvLqrS66uzMJuAlYI2VTkynV19ssB2TtcKU,1336
+ins_pricing/tests/governance/__init__.py,sha256=5Nxg4_dIxY_J58_x2QOXrrRgw6L51Md0Wnt5Up-chqg,39
+ins_pricing/tests/governance/test_audit.py,sha256=ubybXSTVILPN4VxQ2fMnG6oPNv4LjJJE3EsQ53NYdLU,1702
+ins_pricing/tests/governance/test_registry.py,sha256=TvkNMLHViNuopjjho6oETwZ9d6MNaNM1xbL6URPDKSk,4602
+ins_pricing/tests/governance/test_release.py,sha256=Cdo6prZ0xlioAP2AYHodzgASEIa6ZCLjbXW9Me2RGKk,2347
+ins_pricing/tests/modelling/conftest.py,sha256=0KUXnkTgIGEIsf0J4uzIx5Kq4JkDyFo81Mv0qvIzW9k,180
+ins_pricing/tests/modelling/test_cross_val_generic.py,sha256=iLZOFmdyrycB15lFWoQphkFlEjzZTozQXTLVOHLw2Qg,1721
+ins_pricing/tests/modelling/test_distributed_utils.py,sha256=9cStpDw7jPdQwmm0Po-G2tB04uzSR1CoOUZMLuB61yI,466
+ins_pricing/tests/modelling/test_explain.py,sha256=NZqKYuL-eSf9eC5ttjcMirfOc48ORMXZQm5PjmYk4jY,1445
+ins_pricing/tests/modelling/test_geo_tokens_split.py,sha256=Ti57IynKB0aWOosr0g2RcZlDZyZC012XZ-hm6tWXBJg,1554
+ins_pricing/tests/modelling/test_graph_cache.py,sha256=QEI5cLLtQ9_zwRR50KqUf8qxo9Jcp1WLxIGs4dSoMNk,821
+ins_pricing/tests/modelling/test_plotting.py,sha256=4gJax72l40fQrjyJQLOgUmaT6xn6zXpujEaFNeHVwGw,1911
+ins_pricing/tests/modelling/test_plotting_library.py,sha256=SB5RjKTaPydK848V0xpqEaJtEWhRv6ZfnHmnnzjaPh4,4079
+ins_pricing/tests/modelling/test_preprocessor.py,sha256=FqbKltV803Pd-ZY1xBc4XF1T-INDuUliaVcMIDPmBxI,1438
+ins_pricing/tests/pricing/__init__.py,sha256=SVfgUYBlCmc4wjYLMRX5xPFgQZxTS3aHBOA_Cx1aJg4,36
+ins_pricing/tests/pricing/test_calibration.py,sha256=hLZuSWOH4t9WKcQ-2srvYp4P5ldr1Yh1dhl7s61vMp8,2420
+ins_pricing/tests/pricing/test_exposure.py,sha256=CrpSncVce-PGt2XzjOX6qV0SA22vKPUv1u8RlKQjt_g,2054
+ins_pricing/tests/pricing/test_factors.py,sha256=NTE7lz1RWChhoRt2K5003DoNRqG_Gu4X1Aauy2NexOg,5093
+ins_pricing/tests/pricing/test_rate_table.py,sha256=ICHfAQsC9TaxXbQVKM5AvBaJXRTVY723Vaz1XOWNMW8,1250
+ins_pricing/tests/production/__init__.py,sha256=WFWlvBVdjg-E-nKaiJ8VTKNELYufJusufpij1p1xwso,39
+ins_pricing/tests/production/test_inference.py,sha256=l1wxpuukgdr4DLO_pSwyqxByIoVHQHmPR6LYIMQmeE0,2417
+ins_pricing/tests/production/test_monitoring.py,sha256=jettbaVLH4a3efLWeiQ6FFukGEw7mmz6_AeYhYX0caQ,11409
+ins_pricing/tests/production/test_preprocess.py,sha256=tsHYANwJjNlaSo8O4qiwqBvMOMtwmtZymRFm6UrODrE,11084
+ins_pricing/tests/production/test_scoring.py,sha256=fKz2tJomodrRt333apCrjtyJCwg9RHRbWm0lvcU6xm0,9848
+ins_pricing/utils/__init__.py,sha256=Rt5ObBJU--Dy4TA5-HbUZQ2_9BOl5-J3ZhSQovXQ2dw,3641
+ins_pricing/utils/device.py,sha256=6G4BuQowmjwPPHOZepU-o4IJA9vipGDBMBa3vF1Z5P0,8460
+ins_pricing/utils/features.py,sha256=_uPkX6wpzdC3XTeAYSI1qLYDVTN1Lcfh4OJ8OtBw5-8,1704
+ins_pricing/utils/io.py,sha256=uu7PqXVbcvfSnN3ry6PsCKCaYy6kxL3GLXuJRciVDWA,2678
+ins_pricing/utils/logging.py,sha256=XGM8swSRItNR1Qghd36QfojOMKj9AX-lIt-LFgcljvA,3230
+ins_pricing/utils/losses.py,sha256=u7JFgjnu9kuGD4tB8Lgvfihit5lOL0mbExN2h_JV5mg,4008
+ins_pricing/utils/metrics.py,sha256=hh4PeDBYh3cXiOuTq6e6B084Lz1WbrzDvdJ85JRkx6A,13675
+ins_pricing/utils/numerics.py,sha256=gLz3MVaTzz3xyH_qKaONRslBzUdJ7RCY7JoRGWbwa-U,2207
+ins_pricing/utils/paths.py,sha256=ds_nt2JmP4Gq8nERJUdUFjh3rxGL2duOXNMifgutI6g,9153
+ins_pricing/utils/profiling.py,sha256=dwbJIEgZFr-Ow0-F1KT94fBfNev1g2HT_UWqJQ2FCvc,11265
+ins_pricing/utils/torch_compat.py,sha256=UrRsqx2qboDG8WE0OmxNOi08ojwE-dCxTQh0N2s3Rgw,2441
+ins_pricing/utils/validation.py,sha256=4Tw9VUJPk0N-WO3YUqZP-xXRl1Xpubkm0vi3WzzZrv4,13348
+ins_pricing-0.5.1.dist-info/METADATA,sha256=xsczDkdeHh7VLt5cFlsZ9MwA5agobSC12PBvsxu-Tp0,5891
+ins_pricing-0.5.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ins_pricing-0.5.1.dist-info/top_level.txt,sha256=haZuNQpHKNBEPZx3NjLnHp8pV3I_J9QG8-HyJn00FA0,12
+ins_pricing-0.5.1.dist-info/RECORD,,

ins_pricing/modelling/core/BayesOpt.py

@@ -1,146 +0,0 @@
-from __future__ import annotations
-
-import torch
-
-try:
-    from .bayesopt.config_preprocess import (
-        BayesOptConfig,
-        DatasetPreprocessor,
-        OutputManager,
-        VersionManager,
-    )
-    from .bayesopt.core import BayesOptModel
-    from .bayesopt.models import (
-        FeatureTokenizer,
-        FTTransformerCore,
-        FTTransformerSklearn,
-        GraphNeuralNetSklearn,
-        MaskedTabularDataset,
-        ResBlock,
-        ResNetSequential,
-        ResNetSklearn,
-        ScaledTransformerEncoderLayer,
-        SimpleGraphLayer,
-        SimpleGNN,
-        TabularDataset,
-    )
-    from .bayesopt.trainers import (
-        FTTrainer,
-        GLMTrainer,
-        GNNTrainer,
-        ResNetTrainer,
-        TrainerBase,
-        XGBTrainer,
-        _xgb_cuda_available,
-    )
-    from .bayesopt.utils import (
-        EPS,
-        DistributedUtils,
-        IOUtils,
-        PlotUtils,
-        TorchTrainerMixin,
-        TrainingUtils,
-        compute_batch_size,
-        csv_to_dict,
-        ensure_parent_dir,
-        free_cuda,
-        infer_factor_and_cate_list,
-        plot_dlift_list,
-        plot_lift_list,
-        set_global_seed,
-        split_data,
-        tweedie_loss,
-    )
-except ImportError:  # pragma: no cover
-    from bayesopt.config_preprocess import (
-        BayesOptConfig,
-        DatasetPreprocessor,
-        OutputManager,
-        VersionManager,
-    )
-    from bayesopt.core import BayesOptModel
-    from bayesopt.models import (
-        FeatureTokenizer,
-        FTTransformerCore,
-        FTTransformerSklearn,
-        GraphNeuralNetSklearn,
-        MaskedTabularDataset,
-        ResBlock,
-        ResNetSequential,
-        ResNetSklearn,
-        ScaledTransformerEncoderLayer,
-        SimpleGraphLayer,
-        SimpleGNN,
-        TabularDataset,
-    )
-    from bayesopt.trainers import (
-        FTTrainer,
-        GLMTrainer,
-        GNNTrainer,
-        ResNetTrainer,
-        TrainerBase,
-        XGBTrainer,
-        _xgb_cuda_available,
-    )
-    from bayesopt.utils import (
-        EPS,
-        DistributedUtils,
-        IOUtils,
-        PlotUtils,
-        TorchTrainerMixin,
-        TrainingUtils,
-        compute_batch_size,
-        csv_to_dict,
-        ensure_parent_dir,
-        free_cuda,
-        infer_factor_and_cate_list,
-        plot_dlift_list,
-        plot_lift_list,
-        set_global_seed,
-        split_data,
-        tweedie_loss,
-    )
-
-__all__ = [
-    "BayesOptConfig",
-    "DatasetPreprocessor",
-    "OutputManager",
-    "VersionManager",
-    "BayesOptModel",
-    "FeatureTokenizer",
-    "FTTransformerCore",
-    "FTTransformerSklearn",
-    "GraphNeuralNetSklearn",
-    "MaskedTabularDataset",
-    "ResBlock",
-    "ResNetSequential",
-    "ResNetSklearn",
-    "ScaledTransformerEncoderLayer",
-    "SimpleGraphLayer",
-    "SimpleGNN",
-    "TabularDataset",
-    "FTTrainer",
-    "GLMTrainer",
-    "GNNTrainer",
-    "ResNetTrainer",
-    "TrainerBase",
-    "XGBTrainer",
-    "_xgb_cuda_available",
-    "EPS",
-    "DistributedUtils",
-    "IOUtils",
-    "PlotUtils",
-    "TorchTrainerMixin",
-    "TrainingUtils",
-    "compute_batch_size",
-    "csv_to_dict",
-    "ensure_parent_dir",
-    "free_cuda",
-    "infer_factor_and_cate_list",
-    "plot_dlift_list",
-    "plot_lift_list",
-    "set_global_seed",
-    "split_data",
-    "tweedie_loss",
-    "torch",
-]

ins_pricing/modelling/core/__init__.py

@@ -1 +0,0 @@
-"""Core modelling modules (bayesopt + evaluation)."""

ins_pricing/modelling/core/bayesopt/trainers/__init__.py

@@ -1,19 +0,0 @@
-"""Trainer implementations split by model type."""
-from __future__ import annotations
-
-from .trainer_base import TrainerBase
-from .trainer_ft import FTTrainer
-from .trainer_glm import GLMTrainer
-from .trainer_gnn import GNNTrainer
-from .trainer_resn import ResNetTrainer
-from .trainer_xgb import XGBTrainer, _xgb_cuda_available
-
-__all__ = [
-    "TrainerBase",
-    "FTTrainer",
-    "GLMTrainer",
-    "GNNTrainer",
-    "ResNetTrainer",
-    "XGBTrainer",
-    "_xgb_cuda_available",
-]

ins_pricing/modelling/core/bayesopt/utils/__init__.py

@@ -1,86 +0,0 @@
-"""Backward compatibility re-exports from refactored utils modules.
-
-This module ensures all existing imports continue to work:
-    from ins_pricing.modelling.core.bayesopt.utils import EPS, IOUtils, ...
-
-The utils.py file has been split into focused modules for better maintainability:
-- constants.py: EPS, set_global_seed, etc.
-- io_utils.py: IOUtils for file I/O
-- distributed_utils.py: DistributedUtils, TrainingUtils for DDP
-- torch_trainer_mixin.py: TorchTrainerMixin for PyTorch training
-- metrics_and_devices.py: Metrics, GPU/device management, CV strategies, plotting
-"""
-
-from __future__ import annotations
-
-# Constants and simple utilities
-from .constants import (
-    EPS,
-    set_global_seed,
-    ensure_parent_dir,
-    compute_batch_size,
-    tweedie_loss,
-    infer_factor_and_cate_list,
-)
-
-# I/O utilities
-from .io_utils import (
-    IOUtils,
-    csv_to_dict,
-)
-
-# Distributed training
-from .distributed_utils import (
-    DistributedUtils,
-    TrainingUtils,
-    free_cuda,
-)
-
-# PyTorch training mixin
-from .torch_trainer_mixin import (
-    TorchTrainerMixin,
-)
-
-# Metrics, devices, CV, and plotting
-from .metrics_and_devices import (
-    get_logger,
-    MetricFactory,
-    GPUMemoryManager,
-    DeviceManager,
-    CVStrategyResolver,
-    PlotUtils,
-    split_data,
-    plot_lift_list,
-    plot_dlift_list,
-    _OrderedSplitter,
-)
-
-__all__ = [
-    # Constants
-    'EPS',
-    'set_global_seed',
-    'ensure_parent_dir',
-    'compute_batch_size',
-    'tweedie_loss',
-    'infer_factor_and_cate_list',
-    # I/O
-    'IOUtils',
-    'csv_to_dict',
-    # Distributed
-    'DistributedUtils',
-    'TrainingUtils',
-    'free_cuda',
-    # PyTorch
-    'TorchTrainerMixin',
-    # Utilities
-    'get_logger',
-    'MetricFactory',
-    'GPUMemoryManager',
-    'DeviceManager',
-    'CVStrategyResolver',
-    'PlotUtils',
-    'split_data',
-    'plot_lift_list',
-    'plot_dlift_list',
-    '_OrderedSplitter',
-]

ins_pricing/modelling/core/bayesopt/utils/constants.py

@@ -1,183 +0,0 @@
-"""Core constants and simple utility functions.
-
-This module contains:
-- EPS constant for numerical stability
-- set_global_seed() for reproducibility
-- ensure_parent_dir() for file operations
-- compute_batch_size() for adaptive batching
-- tweedie_loss() for regression loss
-- infer_factor_and_cate_list() for automatic feature detection
-"""
-
-from __future__ import annotations
-
-import random
-from pathlib import Path
-from typing import List, Optional, Tuple
-
-import numpy as np
-import pandas as pd
-import torch
-
-# Constants
-# =============================================================================
-EPS = 1e-8
-"""Small epsilon value for numerical stability."""
-
-
-# Simple utility functions
-# =============================================================================
-
-def set_global_seed(seed: int) -> None:
-    """Set random seed for reproducibility across all libraries.
-
-    Args:
-        seed: Random seed value
-    """
-    random.seed(seed)
-    np.random.seed(seed)
-    torch.manual_seed(seed)
-    if torch.cuda.is_available():
-        torch.cuda.manual_seed_all(seed)
-
-
-def ensure_parent_dir(file_path: str) -> None:
-    """Create parent directories when missing.
-
-    Args:
-        file_path: Path to file whose parent directory should be created
-    """
-    directory = Path(file_path).parent
-    if directory and not directory.exists():
-        directory.mkdir(parents=True, exist_ok=True)
-
-
-def compute_batch_size(data_size: int, learning_rate: float,
-                      batch_num: int, minimum: int) -> int:
-    """Compute adaptive batch size based on data size and learning rate.
-
-    Args:
-        data_size: Total number of samples
-        learning_rate: Learning rate value
-        batch_num: Target number of batches
-        minimum: Minimum batch size
-
-    Returns:
-        Computed batch size
-    """
-    estimated = int((learning_rate / 1e-4) ** 0.5 *
-                   (data_size / max(batch_num, 1)))
-    return max(1, min(data_size, max(minimum, estimated)))
-
-
-def tweedie_loss(pred, target, p=1.5, eps=1e-6, max_clip=1e6):
-    """Compute Tweedie deviance loss for PyTorch.
-
-    Reference: https://scikit-learn.org/stable/modules/model_evaluation.html
-
-    Args:
-        pred: Predicted values (tensor)
-        target: True values (tensor)
-        p: Tweedie power parameter (1.0-2.0)
-        eps: Small epsilon for numerical stability
-        max_clip: Maximum value for clipping
-
-    Returns:
-        Tweedie negative log-likelihood (tensor)
-    """
-    # Clamp predictions to positive values for stability
-    pred_clamped = torch.clamp(pred, min=eps)
-
-    if p == 1:
-        # Poisson
-        term1 = target * torch.log(target / pred_clamped + eps)
-        term2 = -target + pred_clamped
-        term3 = 0
-    elif p == 0:
-        # Gaussian
-        term1 = 0.5 * torch.pow(target - pred_clamped, 2)
-        term2 = 0
-        term3 = 0
-    elif p == 2:
-        # Gamma
-        term1 = torch.log(pred_clamped / target + eps)
-        term2 = -target / pred_clamped + 1
-        term3 = 0
-    else:
-        # General Tweedie
-        term1 = torch.pow(target, 2 - p) / ((1 - p) * (2 - p))
-        term2 = target * torch.pow(pred_clamped, 1 - p) / (1 - p)
-        term3 = torch.pow(pred_clamped, 2 - p) / (2 - p)
-
-    return torch.nan_to_num(
-        2 * (term1 - term2 + term3),
-        nan=eps,
-        posinf=max_clip,
-        neginf=-max_clip
-    )
-
-
-def infer_factor_and_cate_list(
-    train_df: pd.DataFrame,
-    test_df: pd.DataFrame,
-    resp_nme: str,
-    weight_nme: str,
-    binary_resp_nme: Optional[str] = None,
-    factor_nmes: Optional[List[str]] = None,
-    cate_list: Optional[List[str]] = None,
-    infer_categorical_max_unique: int = 50,
-    infer_categorical_max_ratio: float = 0.05
-) -> Tuple[List[str], List[str]]:
-    """Infer factor_nmes/cate_list when feature names are not provided.
-
-    Rules:
-      - factor_nmes: start from shared train/test columns, exclude target/weight/(optional binary target).
-      - cate_list: object/category/bool plus low-cardinality integer columns.
-      - Always intersect with shared train/test columns to avoid mismatches.
-
-    Args:
-        train_df: Training DataFrame
-        test_df: Test DataFrame
-        resp_nme: Response/target column name
-        weight_nme: Sample weight column name
-        binary_resp_nme: Optional binary response column name
-        factor_nmes: Optional list of feature column names
-        cate_list: Optional list of categorical feature names
-        infer_categorical_max_unique: Max unique values for categorical inference
-        infer_categorical_max_ratio: Max ratio of unique/total for categorical inference
-
-    Returns:
-        Tuple of (factor_nmes, cate_list)
-    """
-    excluded = {resp_nme, weight_nme}
-    if binary_resp_nme:
-        excluded.add(binary_resp_nme)
-
-    common_cols = [c for c in train_df.columns if c in test_df.columns]
-    if factor_nmes is None:
-        factors = [c for c in common_cols if c not in excluded]
-    else:
-        factors = [
-            c for c in factor_nmes if c in common_cols and c not in excluded
-        ]
-
-    if cate_list is not None:
-        cats = [c for c in cate_list if c in factors]
-        return factors, cats
-
-    n_rows = max(1, len(train_df))
-    cats: List[str] = []
-    for col in factors:
-        s = train_df[col]
-        if (pd.api.types.is_bool_dtype(s) or
-            pd.api.types.is_object_dtype(s) or
-            isinstance(s.dtype, pd.CategoricalDtype)):
-            cats.append(col)
-            continue
-        if pd.api.types.is_integer_dtype(s):
-            nunique = int(s.nunique(dropna=True))
-            if (nunique <= infer_categorical_max_unique or
-                (nunique / n_rows) <= infer_categorical_max_ratio):
-                cats.append(col)
-
-    return factors, cats