Przeglądaj źródła

Fixed some imports and renamed arguments to argument

Christoph Theiß 6 lat temu
rodzic
commit
14e1a79a86

+ 1 - 1
cvargparse/__init__.py

@@ -1,6 +1,6 @@
 __version__ = "0.1.2"
 
-from .arguments import Argument, FileArgument
+from .argument import Argument, FileArgument
 from .factory import ArgFactory
 from .parser import BaseParser, GPUParser
 Arg = Argument

+ 0 - 0
cvargparse/arguments.py → cvargparse/argument.py


+ 4 - 4
cvargparse/factory.py

@@ -1,5 +1,5 @@
-from cvargparse.utils import factory 
-from cvargparse.arguments import Argument as Arg
+from cvargparse.utils import factory
+from cvargparse.argument import Argument as Arg
 
 from abc import ABC
 
@@ -18,11 +18,11 @@ class BaseFactory(ABC):
 
 	def get(self):
 		return self.args
- 
+
 
 class ArgFactory(BaseFactory):
 	'''
-	
+
 	'''
 	@factory
 	def batch_size(self):

+ 2 - 1
cvargparse/parser.py

@@ -1,6 +1,7 @@
 import argparse, logging
 
-from arguments import Argument, FileArgument
+from cvargparse.argument import Argument as Arg
+from cvargparse.factory import ArgFactory
 
 class BaseParser(argparse.ArgumentParser):
 	def __init__(self, arglist=[], nologging=False, sysargs=None, *args, **kw):