cnvrg 0.0.14252 → 0.0.14253

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a6409c6defc4d36177be66185f8dfe3f274f0fc2
-  data.tar.gz: 24e0007c49e1b91ee5d11a6a7874f3af42d0e538
+  metadata.gz: 29e42b6a2a440d51e52338b3a8ed9d58f5533634
+  data.tar.gz: f1c4bb6e33f8c5c0d41a5ca27e70a4b0e668199e
 SHA512:
-  metadata.gz: 541d9fb061b83b0897bea7c58c3b3160b2ec23a43197103f01499c8cbc179d9d466bb8b7e7ebeb670973eade492f5c9481c9d4f39ebc6eff89c006d2eb231f5d
-  data.tar.gz: 48b70dfcc850585b389054e862dbfe6519969a679cc9f0ce1acf29f6c9158aeda29d4dee813ce4b1eeeb841fed8b08e431ef3e71437507a1447ad3eb6a4e2c7c
+  metadata.gz: 6342336d3ed35832f1af2a6dcb6866d48767b678734ae5fd301735f61d9c8b03f01aafd2c50896563da859f4aa3c00d0f587049483223b5b8f66199aaf9db443
+  data.tar.gz: cecc0bd1e7a4df55e4ce78290bc043289967869e764aa9615c720b4ac03a3fc90885e21d4ea2aaf508326e2c47ceb154111121b5767ee97a0a411dc0a88e818c

data/lib/cnvrg/cli.rb

@@ -203,12 +203,16 @@ module Cnvrg
         end
         config = YAML.load_file(home_dir+"/.cnvrg/config.yml")
         owner = config.to_h[:owner]
+        compression_path =  "#{File.expand_path('~')}/.cnvrg/tmp"
 
         say "Setting default api to be: #{url}", Thor::Shell::Color::BLUE
         if config.empty?
-          config = {owner: "", username: "", version_last_check: get_start_day(), api: url}
+          config = {owner: "", username: "", version_last_check: get_start_day(), api: url,compression_path: compression_path }
         else
-          config = {owner: config.to_h[:owner], username: config.to_h[:username], version_last_check: config.to_h[:version_last_check], api: url}
+          if !config.to_h[:compression_path].nil?
+            compression_path = config.to_h[:compression_path]
+          end
+          config = {owner: config.to_h[:owner], username: config.to_h[:username], version_last_check: config.to_h[:version_last_check], api: url, compression_path: compression_path}
         end
         res = Cnvrg::API.request("/users/#{owner}/custom_api", 'POST', {custom_api: url})
         if Cnvrg::CLI.is_response_success(res,false)
@@ -244,7 +248,9 @@ module Cnvrg
           FileUtils.touch [home_dir+"/.cnvrg/config.yml"]
         end
         config = YAML.load_file(home_dir+"/.cnvrg/config.yml")
-        config = {owner: owner, username: current_user, version_last_check: get_start_day(), api: url}
+
+        compression_path =  "#{home_dir}/.cnvrg/tmp"
+        config = {owner: owner, username: current_user, version_last_check: get_start_day(), api: url,compression_path:compression_path}
         File.open(home_dir+"/.cnvrg/config.yml", "w+") { |f| f.write config.to_yaml }
         say "Done"
       rescue
@@ -319,7 +325,40 @@ module Cnvrg
       end
     end
 
-    map %w(--set-default-owner) => :set_default_owner
+    desc 'set_compression_path', 'set compression path'
+    method_option :reset, :type => :boolean, :aliases => ["-r","--reset"], :default => false
+
+    def set_compression_path(*compression_path)
+      begin
+        if (compression_path.nil? or compression_path.empty?) and options["reset"]
+          compression_path =  ["#{File.expand_path('~')}/.cnvrg/tmp"]
+        end
+        compression_path = compression_path.join(" ")
+        if !Dir.exist? compression_path
+          say "Couldn't find #{compression_path}, please make sure it exist", Thor::Shell::Color::RED
+          exit(0)
+        end
+
+        home_dir = File.expand_path('~')
+        path = "#{home_dir}/.cnvrg/config.yml"
+        if !File.exist?(path)
+          say "Couldn't find ~/.cnvrg/config.yml file, please logout and login again", Thor::Shell::Color::RED
+
+          exit(0)
+        end
+        config = YAML.load_file(path)
+        config_new = {owner: config.to_h[:owner], username: config.to_h[:username],
+                  version_last_check: config.to_h[:version_last_check], api: config.to_h[:api], compression_path: compression_path}
+        File.open(home_dir+"/.cnvrg/config.yml", "w+") { |f| f.write config_new.to_yaml }
+        checks = Helpers.checkmark
+        say "#{checks} Done", Thor::Shell::Color::GREEN
+
+      rescue SignalException
+        say "\nAborting"
+        exit(1)
+      end
+    end
+
 
 
     desc 'login', 'Authenticate with cnvrg.io platform'
@@ -1182,7 +1221,8 @@ module Cnvrg
           say "Compressing data", Thor::Shell::Color::BLUE
 
           home_dir = File.expand_path('~')
-          tar_path = "#{home_dir}/.cnvrg/tmp/#{@dataset.slug}_#{commit_sha1}.tar.gz"
+          compression_path = get_compression_path
+          tar_path = "#{compression_path}#{@dataset.slug}_#{commit_sha1}.tar.gz"
           tar_files_path = "#{home_dir}/.cnvrg/tmp/#{@dataset.slug}_#{commit_sha1}.txt"
           tar_files = (result["added"] + result["updated_on_local"]).join("\n")
           File.open(tar_files_path, 'w') { |f| f.write tar_files }
@@ -1546,6 +1586,8 @@ module Cnvrg
         if result["added"].size > 0
           FileUtils.rm_rf(result["added"])
         end
+        say "Changes were removed successfully", Thor::Shell::Color::GREEN
+
 
       rescue SignalException
         log_end(-1)
@@ -2152,7 +2194,6 @@ module Cnvrg
     method_option :data, :type => :string, :aliases => ["-d", "--data"], :default => ""
     method_option :data_commit, :type => :string, :aliases => ["-dc", "--data_commit"], :default => ""
     method_option :ignore, :type => :string, :aliases => ["-i", "--ignore"], :desc => "ignore following files", :default => ""
-
     def exec(*cmd)
 
       log = []
@@ -2580,7 +2621,8 @@ module Cnvrg
     method_option :data_commit, :type => :string, :aliases => ["--data_commit"], :default => ""
     method_option :ignore, :type => :string, :aliases => ["-i", "--ignore"], :desc => "ignore following files", :default => ""
 
-    def exec_remote(*cmd)
+    def
+    exec_remote(*cmd)
       verify_logged_in(true)
       log_start(__method__, args, options)
       working_dir = is_cnvrg_dir
@@ -3905,7 +3947,7 @@ module Cnvrg
         #     },
         # }
 
-        container_id = `nvidia-docker run -itd -p #{port}:8888 -p #{tensport}:6006 -w #{app_dir} #{image_name}:latest #{cmd} `
+        container_id = `nvidia-docker run -itd -p #{port}:8888 -p #{tensport}:6006 -w #{app_dir} -v /usr/bin/nvidia-smi:/usr/bin/nvidia-smi #{image_name}:latest #{cmd} `
         container_id = container_id.gsub("\n", "")
         container = Docker::Container.get(container_id)
         # container.start()
@@ -4402,7 +4444,13 @@ module Cnvrg
           if url.nil? or url.empty?
             url = "https://cnvrg.io/api"
           end
-          config = {owner: owner, username: username, version_last_check: get_start_day(), api: url}
+          config = YAML.load_file(home_dir+"/.cnvrg/config.yml")
+
+          compression_path =  "#{home_dir}/.cnvrg/tmp"
+            if !config.empty? and !config.to_h[:compression_path].nil?
+              compression_path = config.to_h[:compression_path]
+            end
+          config = {owner: owner, username: username, version_last_check: get_start_day(), api: url,compression_path:compression_path}
 
           File.open(home_dir+"/.cnvrg/config.yml", "w+") { |f| f.write config.to_yaml }
           return true
@@ -4510,6 +4558,19 @@ module Cnvrg
         api = config.to_h[:api]
         return api.gsub!("/api", "")
       end
+      def  get_compression_path
+        home_dir = File.expand_path('~')
+
+        config = YAML.load_file(home_dir+ "/.cnvrg/config.yml")
+        compression_path = config.to_h[:compression_path]
+        if compression_path.nil?
+          compression_path = "#{home_dir}/.cnvrg/tmp/"
+        end
+        if !compression_path.ends_with? "/"
+          compression_path = compression_path + "/"
+        end
+        return compression_path
+      end
 
       def get_project_home
         absolute_path = Dir.pwd

data/lib/cnvrg/version.rb

@@ -1,5 +1,4 @@
 module Cnvrg
-    VERSION = '0.0.14252'
-
+    VERSION = '0.0.14253'
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cnvrg
 version: !ruby/object:Gem::Version
-  version: 0.0.14252
+  version: 0.0.14253
 platform: ruby
 authors:
 - Yochay Ettun
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-06-29 00:00:00.000000000 Z
+date: 2017-07-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -373,8 +373,9 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.12
+rubygems_version: 2.5.1
 signing_key: 
 specification_version: 4
 summary: A CLI tool for interacting with cnvrg.io.
 test_files: []
+has_rdoc: