Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def action_info(argc, argv):
args = parse_args(argv)
if args.to_json is True:
devices = []
info = {
"version": __version__,
"keras_version": keras.__version__,
"tf_version": tf.__version__,
'sklean_version': sklearn.__version__,
"devices":[]
}
devs = device_lib.list_local_devices()
for dev in devs:
info['devices'].append({
'name': dev.name,
'type': dev.device_type,
'memory': dev.memory_limit,
'description': dev.physical_device_desc
})
print(json.dumps(info))
else:
from ergo.version import __version__, __author__, __email__, __license__
import os
desc = 'ergo is a tool that makes deep learning with Keras easier.'
required = []
with open('requirements.txt') as fp:
for line in fp:
line = line.strip()
if line != "":
required.append(line)
setup( name = 'ergo-ai',
version = __version__,
description = desc,
long_description = desc,
long_description_content_type = 'text/plain',
author = __author__,
author_email = __email__,
url = 'http://www.github.com/evilsocket/ergo',
packages = find_packages(),
install_requires = required,
scripts = [ 'bin/ergo' ],
license = __license__,
classifiers = [
'Programming Language :: Python :: 3',
'Development Status :: 5 - Production/Stable',
'License :: OSI Approved :: GNU General Public License (GPL)',
'Environment :: Console',
],