save pickle in managed folder

Isma
Level 1
save pickle in managed folder

Hello,

I want to save a picke object in DSS folder. Here is an example of working code outside DSS. How can I use the DSS API to do so in DSS.

def save_model(self๐Ÿ˜ž
# Names
scope=self.scope.replace(' ', '_').lower()
model_name='model_' + scope
opt_tree_name='opt_tree_' + scope
pkl_name=model_name + '.pkl'
# Save pickles
with open(self.builds_path + pkl_name, 'wb') as f:
pickle.dump(self, f)
0 Kudos
2 Replies
AlexandreL
Dataiker

Hi,

You can use the folder api to get a writer:

with dataiku.Folder(FOLDER_ID).get_writer(MODEL_PATH_IN_FOLDER) as writer:
	writeable = pickle.dumps(MODEL)
	writer.write(writeable)

 

ssuhas76
Level 3

What if I need to do this in a endpioint code which runs outside of dataiku. I will have to used dataikuapi right? Do you know how this can be done?

0 Kudos