diff --git a/setup.py b/setup.py index 4434d66..10185c1 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ tests_require=test_requirements, entry_points={ 'console_scripts': [ - 'angle-trainer = angle_emb.train_cli:__main__', + 'angle-trainer = angle_emb.train_cli:main', ], }, )