Correct ferminet main program to have correct shebang

PiperOrigin-RevId: 765210239
Change-Id: Ifcbf27f0ef293ffc969308f49a9fc3a97e16c212

Author: jsspencer

README.md

@@ -56,6 +56,12 @@ calculation.
 ferminet --config ferminet/configs/atom.py --config.system.atom Li --config.batch_size 256 --config.pretrain.iterations 100
 ```
 
+or
+
+```shell
+python3 ferminet/main.py --config ferminet/configs/atom.py --config.system.atom Li --config.batch_size 256 --config.pretrain.iterations 100
+```
+
 will train FermiNet to find the ground-state wavefunction of the Li atom using a
 batch size of 1024 MCMC configurations ("walkers" in variational Monte Carlo
 language), and 100 iterations of pretraining (the default of 1000 is overkill

bin/ferminet ferminet/main.pybin/ferminet renamed to ferminet/main.py

@@ -35,5 +35,10 @@ def main(_):
   train.train(cfg)
 
 
+def main_wrapper():
+  # For calling from setuptools' console_script entry-point.
+  app.run(main)
+
+
 if __name__ == '__main__':
   app.run(main)

setup.py

@@ -49,13 +49,19 @@ def ferminet_test_suite():
 setup(
     name='ferminet',
     version='0.2',
-    description=('A library to train networks to represent ground '
-                 'state wavefunctions of fermionic systems'),
+    description=(
+        'A library to train networks to represent ground '
+        'state wavefunctions of fermionic systems'
+    ),
     url='https://github.com/deepmind/ferminet',
     author='DeepMind',
     author_email='no-reply@google.com',
     # Contained modules and scripts.
-    scripts=['bin/ferminet'],
+    entry_points={
+        'console_scripts': [
+            'ferminet = ferminet.main:main_wrapper',
+        ],
+    },
     packages=find_packages(),
     install_requires=REQUIRED_PACKAGES,
     extras_require={'testing': ['flake8', 'pylint', 'pytest', 'pytype']},