From 88161bd114697b0c2d3a1a434c6ad9d05defe21c Mon Sep 17 00:00:00 2001
From: SimpleArt <71458112+SimpleArt@users.noreply.github.com>
Date: Tue, 29 Dec 2020 19:33:46 -0500
Subject: [PATCH] Added function info to decorators

---
 src/termination_point/termination_methods.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/termination_point/termination_methods.py b/src/termination_point/termination_methods.py
index e7920c5..3776f67 100644
--- a/src/termination_point/termination_methods.py
+++ b/src/termination_point/termination_methods.py
@@ -1,3 +1,6 @@
+from EasyGA import function_info
+
+@function_info
 def _add_by_fitness_goal(termination_impl):
     """Adds termination by fitness goal to the method."""
 
@@ -25,10 +28,10 @@ def _add_by_fitness_goal(termination_impl):
         # Check other termination methods
         return termination_impl(ga)
 
-    new_method.__name__ = termination_impl.__name__
     return new_method
 
 
+@function_info
 def _add_by_generation_goal(termination_impl):
     """Adds termination by generation goal to the method."""
 
@@ -41,10 +44,10 @@ def _add_by_generation_goal(termination_impl):
         # Check other termination methods
         return termination_impl(ga)
 
-    new_method.__name__ = termination_impl.__name__
     return new_method
 
 
+@function_info
 def _add_by_tolerance_goal(termination_impl):
     """Adds termination by tolerance goal to the method."""
 
@@ -71,7 +74,6 @@ def _add_by_tolerance_goal(termination_impl):
         # Check other termination methods
         return termination_impl(ga)
 
-    new_method.__name__ = termination_impl.__name__
     return new_method