MLCatchUp: Automated Update of Deprecated Machine-Learning APIs in Python

Stefanus A Haryono (1) , Ferdian Thung (1) , David Lo (1) , Julia Lawall (2) , Lingxiao Jiang (1)

1 SIS - Singapore Management University
2 Whisper - Well Honed Infrastructure Software for Programming Environments and Runtimes

Machine learning (ML) libraries are gaining vast popularity, especially in the Python programming language. Using the latest version of such libraries is recommended to ensure the best performance and security. When migrating to the latest version of a machine learning library, usages of deprecated APIs need to be updated, which is a time-consuming process. In this paper, we propose MLCatchUp, an automated API usage update tool for deprecated APIs of popular ML libraries written in Python. MLCatchUp automatically infers the required transformation to migrate usages of deprecated API through the differences between the deprecated and updated API signatures. MLCatchUp offers a readable transformation rule in the form of a domain specific language (DSL). We evaluate MLCatchUp using a dataset of 267 real-world Python code containing 551 usages of 68 distinct deprecated APIs, where MLCatchUp achieves 90.7% accuracy. A video demonstration of MLCatchUp is available at https://youtu.be/5NjOPNt5iaA.

Mots clés: Python; Program Transformation; Automatic Update; Deprecated API

Domaines: Génie logiciel [cs.SE]

Stefanus A Haryono, Ferdian Thung, David Lo, Julia Lawall, Lingxiao Jiang. MLCatchUp: Automated Update of Deprecated Machine-Learning APIs in Python. ICSME 2021 - 37th IEEE International Conference on Software Maintenance and Evolution, Sep 2021, Luxembourg City / Virtual, Luxembourg. ⟨10.1109/ICSME52107.2021.00061⟩. ⟨hal-03361370⟩