X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=python.git;a=blobdiff_plain;f=poly.py;h=818742bbabcf365ae9aa886e7da4d3cb3a2af783;hp=ae9f88e92b2b1b3caa84f4a612cdac0b0de4cd79;hb=32b7d182b56015b9154f3aefc3d5c71a708ebe8f;hpb=488798e2df8845d1991243d26053eec2d5031b75

diff --git a/poly.py b/poly.py
index ae9f88e..818742b 100755
--- a/poly.py
+++ b/poly.py
@@ -5,6 +5,8 @@
 
 # Written by Francois Fleuret <francois@fleuret.org>
 
+######################################################################
+
 import torch
 
 def pol_prod(a, b):
@@ -22,8 +24,10 @@ def pol_prim(a):
     n[1:] = a / n[1:]
     return n
 
-a = torch.tensor([1., 2., 3.])
-b = torch.tensor([2., 5.])
+######################################################################
 
-print(pol_prod(a, b))
-print(pol_prim(b))
+if __name__ == '__main__':
+    a = torch.tensor([1., 2., 3.])
+    b = torch.tensor([2., 5.])
+    print(pol_prod(a, b))
+    print(pol_prim(b))