Merge pull request #64 from StatCan/add-prob-volv2

fix(use the better label)
StatCan · Sep 23, 2021 · e0a6302 · e0a6302
2 parents da808ea + d9fafe6
commit e0a6302
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks.go b/notebooks.go
@@ -234,7 +234,7 @@ func (s *server) GetNotebooks(w http.ResponseWriter, r *http.Request) {
 func (s *server) handleVolume(ctx context.Context, req volumerequest, notebook *kubeflowv1.Notebook) error {
 	var pvc = corev1.PersistentVolumeClaim{}
 	if req.Type == VolumeTypeNew {
-		if _, ok := notebook.Labels["notebook.statcan.gc.ca/protected-b"]; ok {
+		if notebook.GetObjectMeta().GetLabels()["notebook.statcan.gc.ca/protected-b"] == "true" {
 			pvc = corev1.PersistentVolumeClaim{
 				ObjectMeta: v1.ObjectMeta{
 					Name:      req.Name,