paramflow 0.3.2__tar.gz → 0.3.3__tar.gz

{paramflow-0.3.2/paramflow.egg-info → paramflow-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: paramflow
-Version: 0.3.2
+Version: 0.3.3
 Summary: A lightweight library for hyperparameter and configuration management
 Home-page: https://github.com/mduszyk/paramflow
 Classifier: Programming Language :: Python :: 3

{paramflow-0.3.2 → paramflow-0.3.3}/paramflow/convert.py

@@ -33,3 +33,18 @@ def convert_type(dst_value, src_value, path=''):
         if path != '':
             path += ' '
         raise TypeError(f'unable to convert {path}{src_type} to {dst_type}') from e
+
+
+def infer_type(value: str):
+    result = value
+    if '.' in value:
+        try:
+            result = float(value)
+        except ValueError:
+            pass
+    else:
+        try:
+            result = int(value)
+        except ValueError:
+            pass
+    return result

{paramflow-0.3.2 → paramflow-0.3.3}/paramflow/params.py

@@ -23,7 +23,7 @@ def load(*sources: Tuple[str, ...],
     """
     Load parameters form multiple sources, layer them on top of each other and activate profile.
     Activation of profile means learying it on top of the default profile.
-    :param source: file or multiple files to load parameters from
+    :param sources: file or multiple files to load parameters from
     :param meta_env_prefix: prefix for env vars that are used to overwrite meta params
     :param meta_args_prefix: prefix for command-line arguments to overwrite meta params
     :param env_prefix: prefix for env vars that are used to overwrite params, if None disable auto adding env source
@@ -44,7 +44,8 @@ def load(*sources: Tuple[str, ...],
         '__source__': ['pf.load'],
     }
     meta_env_parser = EnvParser(meta_env_prefix, 'default')
-    meta_args_parser = ArgsParser(meta_args_prefix, 'default', no_exit=True, descr='Meta-parameters')
+    meta_args_parser = ArgsParser(meta_args_prefix, 'default',
+                                  no_exit=True, consume_args=True, descr='Meta-parameters')
     meta = deep_merge(meta, meta_env_parser(meta))
     meta = deep_merge(meta, meta_args_parser(meta))
     meta = freeze(meta)

{paramflow-0.3.2 → paramflow-0.3.3}/paramflow/parser.py

@@ -2,11 +2,14 @@ import argparse
 import configparser
 import json
 import os
+import sys
 from abc import ABC, abstractmethod
 from typing import Dict, Final, Type
 
 import yaml
 
+from paramflow.convert import infer_type
+
 
 class Parser(ABC):
     @abstractmethod
@@ -131,12 +134,13 @@ class NoExitArgumentParser(argparse.ArgumentParser):
 class ArgsParser(Parser):
 
     def __init__(self, prefix: str, default_profile: str, target_profile: str = None,
-                 no_exit: bool = False, descr: str = None):
+                 no_exit: bool = False, descr: str = None, consume_args: bool = False):
         self.prefix = prefix
         self.default_profile = default_profile
         self.target_profile = target_profile
         self.no_exit = no_exit
         self.descr = descr
+        self.consume_args = consume_args
 
     def __call__(self, params: Dict[str, any]) -> Dict[str, any]:
         if self.target_profile is None and self.default_profile in params:
@@ -151,12 +155,21 @@ class ArgsParser(Parser):
             if typ is dict or typ is list or typ is bool or typ is tuple or value is None:
                 typ = str
             parser.add_argument(f'--{self.prefix}{key}', type=typ, default=None, help=f'{key} = {value}')
-        args, _ = parser.parse_known_args()
+        args, remaining = parser.parse_known_args()
         args_params = {}
         for arg_key, arg_value in args.__dict__.items():
             if arg_value is not None:
                 key = arg_key.replace(self.prefix, '')
                 args_params[key] = arg_value
+
+        if self.consume_args:
+            sys.argv = [sys.argv[0]] + remaining
+        else:
+            it = iter(remaining)
+            for arg_name, arg_value in zip(it, it):
+                key = arg_name.replace('--', '').replace(self.prefix, '')
+                args_params[key] = infer_type(arg_value)
+
         result = args_params
         if len(args_params) > 0:
             if self.target_profile is not None:

{paramflow-0.3.2 → paramflow-0.3.3/paramflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: paramflow
-Version: 0.3.2
+Version: 0.3.3
 Summary: A lightweight library for hyperparameter and configuration management
 Home-page: https://github.com/mduszyk/paramflow
 Classifier: Programming Language :: Python :: 3

{paramflow-0.3.2 → paramflow-0.3.3}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='paramflow',
-    version='0.3.2',
+    version='0.3.3',
     description='A lightweight library for hyperparameter and configuration management', 
     packages=find_packages(),
     install_requires=[