소스 검색

Fixed a bug where one could not parse new factories

Christoph Theiß 6 년 전
부모
커밋
3df87cbcc0
2개의 변경된 파일3개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 1
      cvargparse/__init__.py
  2. 2 2
      cvargparse/parser.py

+ 1 - 1
cvargparse/__init__.py

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

+ 2 - 2
cvargparse/parser.py

@@ -1,14 +1,14 @@
 import argparse, logging
 import argparse, logging
 
 
 from cvargparse.argument import Argument as Arg
 from cvargparse.argument import Argument as Arg
-from cvargparse.factory import ArgFactory
+from cvargparse.factory import BaseFactory
 
 
 class BaseParser(argparse.ArgumentParser):
 class BaseParser(argparse.ArgumentParser):
 	def __init__(self, arglist=[], nologging=False, sysargs=None, *args, **kw):
 	def __init__(self, arglist=[], nologging=False, sysargs=None, *args, **kw):
 		super(BaseParser, self).__init__(*args, **kw)
 		super(BaseParser, self).__init__(*args, **kw)
 		self.__nologging = nologging
 		self.__nologging = nologging
 		self.__sysargs = sysargs
 		self.__sysargs = sysargs
-		if isinstance(arglist, ArgFactory):
+		if isinstance(arglist, BaseFactory):
 			arglist = arglist.get()
 			arglist = arglist.get()
 
 
 		for arg in arglist:
 		for arg in arglist: