econmethods 3.0.2__tar.gz → 3.0.3__tar.gz

{econmethods-3.0.2 → econmethods-3.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: econmethods
-Version: 3.0.2
+Version: 3.0.3
 Summary: A python package implementing various econometrical tests and estimators
 Home-page: https://github.com/NaturionBG/econmethods
 Author: NaturionBG

{econmethods-3.0.2 → econmethods-3.0.3}/econmethods/Lib.py

@@ -444,7 +444,10 @@ class FECM:
           flag = True
           for coef, t in zip(res_lr.params.index, res_lr.pvalues):
             if t > self.sf:
-              lr_fe = lr_fe.drop(columns=[coef])
+              if coef != 'const':
+                lr_fe = lr_fe.drop(columns=[coef])
+              else:
+                self.__lr_c = False
               flag=False
           if flag:
             break
@@ -513,7 +516,11 @@ class FECM:
           est.append(sm.OLS(model['target_diff'], sm.add_constant(model.iloc[:, 3:])).fit())
         else:
           est.append(sm.OLS(model['target_diff'], model.iloc[:, 3:]).fit())
-        self.__ccemg_units[i] = self.__ccemg_units[i].drop(columns=drop)
+        if 'const' not in drop:
+          self.__ccemg_units[i] = self.__ccemg_units[i].drop(columns=drop)
+        else:
+          self.__C = False
+          drop = drop.remove('const')
       return est
     elif self.__method == 'mg':
       units = self.__mg_units
@@ -523,7 +530,11 @@ class FECM:
           est.append(sm.OLS(model['target_diff'], sm.add_constant(model.iloc[:, 3:])).fit())
         else:
           est.append(sm.OLS(model['target_diff'], model.iloc[:, 3:]).fit())
-        self.__ccemg_units[i] = self.__ccemg_units[i].drop(columns=drop)
+        if 'const' not in drop:
+          self.__mg_units[i] = self.__mg_units[i].drop(columns=drop)
+        else:
+          self.__C = False
+          drop = drop.remove('const')
       return est
     elif self.__method == 'ccep':
       units = self.get_ccemg_frames(self.__lag)
@@ -538,7 +549,11 @@ class FECM:
             ts = []
             flag = True
             if max(zip(est[0].params.index, est[0].pvalues), key=lambda x: x[1])[1] > self.sf:
-              pool = pool.drop(columns=[max(zip(est[0].params.index, est[0].pvalues), key=lambda x: x[1])[0]])
+              to_drop = max(zip(est[0].params.index, est[0].pvalues), key=lambda x: x[1])[0]
+              if to_drop != 'const':
+                pool = pool.drop(columns=[to_drop])
+              else:
+                self.__C = False
               flag=False
             for i in self.x_difs:
               try:

{econmethods-3.0.2 → econmethods-3.0.3}/econmethods.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: econmethods
-Version: 3.0.2
+Version: 3.0.3
 Summary: A python package implementing various econometrical tests and estimators
 Home-page: https://github.com/NaturionBG/econmethods
 Author: NaturionBG

{econmethods-3.0.2 → econmethods-3.0.3}/setup.py

@@ -5,7 +5,7 @@ with open('README.md', 'r') as f:
   
 setup(
   name = 'econmethods',
-  version = 'v3.0.2',
+  version = 'v3.0.3',
   description='A python package implementing various econometrical tests and estimators',
   packages = find_packages(),
   long_description=long_desc,