1 files changed, 2 insertions, 0 deletions
diff --git a/model.rb b/model.rb
index bab5bde..05075f8 100644
--- a/model.rb
+++ b/model.rb
@@ -19,6 +19,8 @@ class ToxCreateModel < Ohm::Model
 
 	attribute :training_dataset
 	attribute :feature_dataset
+	attribute :endpoint_uri
+	attribute :endpoint
 	#attributey :validation_task_uri
 	attribute :validation_uri